Fix eval code

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