Merge branch 'saaketh/tok_encodings' of github.com-me:mosaicml/llm-fo…

…undry into saaketh/tok_encodings merging main
mosaicml · Jun 6, 2024 · bf95e9e · bf95e9e
2 parents 8a9bb68 + 7c8fb91
commit bf95e9e
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),
         )