fix

mosaicml · Oct 25, 2024 · 8e42217 · 8e42217
1 parent 04ddfaa
commit 8e42217
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -76,6 +76,11 @@ def _maybe_get_license_filename(
 
     If the license file does not exist, returns None.
     """
+    # Early return if no local directory exists
+    if not os.path.exists(local_dir):
+        return None
+
+    # Try to find the license file
     try:
         license_filename = next(
             file for file in os.listdir(local_dir)