fixed precommit.

kohankhaki · kohankhaki · commit 8d88cb12716b · 2025-04-04T09:30:32.000-04:00
diff --git a/vec_inf/cli/_helper.py b/vec_inf/cli/_helper.py
@@ -153,11 +153,6 @@ def build_launch_command(self) -> str:
                 f"{self.params['log_dir']}/{self.model_name}.%j/{self.model_name}.%j.err",
             ]
         )
-        # Add slurm script
-        # slurm_script = "vllm.slurm"
-        # if int(self.params["num_nodes"]) > 1:
-        #     slurm_script = "multinode_vllm.slurm"
-        # command_list.append(f"{SRC_DIR}/{slurm_script}")
 
         slurm_script_path = SlurmScriptGenerator(
             self.params, src_dir=SRC_DIR, is_multinode=int(self.params["num_nodes"]) > 1

Original file line number	Diff line number	Diff line change
`@@ -153,11 +153,6 @@ def build_launch_command(self) -> str:`
`153`	`153`	`f"{self.params['log_dir']}/{self.model_name}.%j/{self.model_name}.%j.err",`
`154`	`154`	`]`
`155`	`155`	`)`
`156`		`- # Add slurm script`
`157`		`- # slurm_script = "vllm.slurm"`
`158`		`- # if int(self.params["num_nodes"]) > 1:`
`159`		`- # slurm_script = "multinode_vllm.slurm"`
`160`		`- # command_list.append(f"{SRC_DIR}/{slurm_script}")`
`161`	`156`
`162`	`157`	`slurm_script_path = SlurmScriptGenerator(`
`163`	`158`	`self.params, src_dir=SRC_DIR, is_multinode=int(self.params["num_nodes"]) > 1`