Skip to content

Commit

Permalink
orig params
Browse files Browse the repository at this point in the history
  • Loading branch information
dakinggg committed Jul 27, 2024
1 parent 799279b commit 354f746
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion llmfoundry/utils/config_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -531,7 +531,6 @@ def process_init_device(model_cfg: Dict[str, Any], fsdp_config: Optional[Dict]):
fsdp_config['sync_module_states'] = True

# Set defaults for mixed initialization
fsdp_config.setdefault('use_orig_params', False)
fsdp_config.setdefault('load_monolith_rank0_only', True)

# Set ffn_config.device_mesh to fsdp_config.device_mesh
Expand Down

0 comments on commit 354f746

Please sign in to comment.