Formatting fix

microsoft · Dec 30, 2024 · d143141 · d143141
1 parent cc1478e
commit d143141
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/deepspeed/checkpoint/ds_to_universal.py b/deepspeed/checkpoint/ds_to_universal.py
@@ -111,7 +111,10 @@ def _save_checkpoint(file_path, chkpt_sd):
 
 def extract_zero_shards(dir, ds_checkpoint, indices_3D):
     pp_index, tp_index, dp_index = indices_3D
-    sd = ds_checkpoint.get_zero_checkpoint_state(pp_index=pp_index, tp_index=tp_index, dp_index=dp_index, strip_tensor_paddings=False)
+    sd = ds_checkpoint.get_zero_checkpoint_state(pp_index=pp_index,
+                                                 tp_index=tp_index,
+                                                 dp_index=dp_index,
+                                                 strip_tensor_paddings=False)
 
     # pprint(f"Processing {dp_index=} {pp_index=}, {tp_index=}")