Merge branch 'master' into lekurile/bloom_v_check

microsoft · Nov 20, 2024 · e4cf472 · e4cf472
2 parents 3e1a054 + 83e4364
commit e4cf472
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/runtime/zero/config.py b/deepspeed/runtime/zero/config.py
@@ -311,7 +311,7 @@ class DeepSpeedZeroConfig(DeepSpeedConfigModel):
     for efficient all_2_all_reduce comm
     """
 
-    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