From 354f746500bf199ca88cbcc04d6feed1a3c4e36e Mon Sep 17 00:00:00 2001 From: Daniel King Date: Sat, 27 Jul 2024 00:20:21 -0700 Subject: [PATCH] orig params --- llmfoundry/utils/config_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/llmfoundry/utils/config_utils.py b/llmfoundry/utils/config_utils.py index 48290bd7c5..dcb97eb0de 100644 --- a/llmfoundry/utils/config_utils.py +++ b/llmfoundry/utils/config_utils.py @@ -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