yo

mosaicml · Jul 29, 2024 · b3b70fc · b3b70fc
1 parent 10b1ea9
commit b3b70fc
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -217,10 +217,13 @@ def __init__(
             )
 
         self.mlflow_logging_config = mlflow_logging_config
-        self.pretrained_model_name = self.mlflow_logging_config['metadata'].get(
-            'pretrained_model_name',
-            None,
-        )
+        if 'metadata' in self.mlflow_logging_config:
+            self.pretrained_model_name = self.mlflow_logging_config['metadata'].get(
+                'pretrained_model_name',
+                None,
+            )
+        else:
+            self.pretrained_model_name = None
 
         self.huggingface_folder_name_fstr = os.path.join(
             'huggingface',