diff --git a/deepspeed/runtime/zero/config.py b/deepspeed/runtime/zero/config.py index 7cac7e3c1ce7..19b272ce9e92 100644 --- 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 diff --git a/setup.py b/setup.py index e39d8c7e05a3..c0452f867b31 100755 --- a/setup.py +++ b/setup.py @@ -202,8 +202,8 @@ def op_enabled(op_name): print(f'Install Ops={install_ops}') # Write out version/git info. -git_hash_cmd = shlex.split("bash -c git rev-parse --short HEAD") -git_branch_cmd = shlex.split("bash -c git rev-parse --abbrev-ref HEAD") +git_hash_cmd = shlex.split("bash -c \"git rev-parse --short HEAD\"") +git_branch_cmd = shlex.split("bash -c \"git rev-parse --abbrev-ref HEAD\"") if command_exists('git') and not is_env_set('DS_BUILD_STRING'): try: result = subprocess.check_output(git_hash_cmd)