update branch names

Author: chibu

examples/guidellm_example.py

@@ -14,7 +14,7 @@
     data_type="emulated",
     max_seconds=30,
     data="prompt_tokens=512,generated_tokens=256,output_tokens=256",
-    branch = "update_guidellm",
+    #branch = "update_guidellm",
     #vllm_kwargs={"enable-chunked-prefill": True}
 )
 

src/automation/tasks/base_task.py

@@ -19,7 +19,8 @@ def __init__(
         packages: Optional[Sequence[str]]=None,
         task_type: str="training",
     ):
-        base_packages = [f"git+https://github.com/neuralmagic/research.git@{branch}"]
+        branch_name = branch or DEFAULT_RESEARCH_BRANCH
+        base_packages = [f"git+https://github.com/neuralmagic/research.git@{branch_name}"]
 
         if packages is not None:
             packages = list(set(packages + base_packages))