Merge branch 'microsoft:master' into LoCo-Zero++

microsoft · Nov 20, 2024 · 11cb119 · 11cb119
2 parents 8cd96ab + 83e4364
commit 11cb119
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/runtime/zero/config.py b/deepspeed/runtime/zero/config.py
@@ -322,7 +322,7 @@ class DeepSpeedZeroConfig(DeepSpeedConfigModel):
     See LoCo paper for more details: (https://arxiv.org/abs/2407.04480).
     """
 
-    mics_shard_size: int = Field(-1, new_param="mics_shard_size")
+    mics_shard_size: int = Field(-1, json_schema_extra={"new_param": "mics_shard_size"})
 
     mics_hierarchical_params_gather: bool = False