Skip to content

Commit a6a090c

Browse files
[pre-commit.ci] Add auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 348a22e commit a6a090c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vec_inf/client/_slurm_script_generator.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,9 @@ def _generate_preamble(self) -> str:
3838

3939
def _generate_shared_args(self) -> str:
4040
if self.is_multinode and not self.params["pipeline_parallelism"]:
41-
tensor_parallel_size = self.params["num_nodes"] * self.params["gpus_per_node"]
41+
tensor_parallel_size = (
42+
self.params["num_nodes"] * self.params["gpus_per_node"]
43+
)
4244
pipeline_parallel_size = 1
4345
else:
4446
tensor_parallel_size = self.params["gpus_per_node"]

0 commit comments

Comments
 (0)