Fix MPT HF conversion (#1257)

mosaicml · Jun 6, 2024 · 67928cb · 67928cb
1 parent ac56dc5
commit 67928cb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/llmfoundry/utils/huggingface_hub_utils.py b/llmfoundry/utils/huggingface_hub_utils.py
@@ -280,6 +280,9 @@ def edit_files_for_hf_compatibility(
         for f in files_processed_and_queued
     }
     for entrypoint in entrypoint_files:
+        file_path = os.path.join(folder, entrypoint)
+        if not os.path.exists(file_path):
+            continue
         existing_relative_imports = get_all_relative_imports(
             os.path.join(folder, entrypoint),
         )