Fix typo in eval code by using 'fsdp' instead of 'fsdp_config' (#1593)

mosaicml · Oct 15, 2024 · de8de10 · de8de10
1 parent 18a28f6
commit de8de10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/command_utils/eval.py b/llmfoundry/command_utils/eval.py
@@ -110,7 +110,7 @@ def evaluate_model(
             mosaicml_logger._flush_metadata(force_flush=True)
 
     fsdp_config = parallelism_config.get(
-        'fsdp_config',
+        'fsdp',
         None,
     ) if parallelism_config else None
     if fsdp_config and model.get('load_in_8bit', False):