precommit

mosaicml · Sep 16, 2024 · e1aefa5 · e1aefa5
1 parent a108aaf
commit e1aefa5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llmfoundry/command_utils/eval.py b/llmfoundry/command_utils/eval.py
@@ -100,7 +100,8 @@ def evaluate_model(
             mosaicml_logger._flush_metadata(force_flush=True)
 
     fsdp_config = parallelism_config.get(
-        'fsdp_config', None
+        'fsdp_config',
+        None,
     ) if parallelism_config else None
     if fsdp_config and model.get('load_in_8bit', False):
         raise ValueError(