Avoid circular import in hf checkpointer (#1304)

dakinggg · web-flow · commit 2412b591d898 · 2024-06-26T00:32:11.000Z
diff --git a/llmfoundry/utils/huggingface_hub_utils.py b/llmfoundry/utils/huggingface_hub_utils.py
@@ -279,7 +279,7 @@ def edit_files_for_hf_compatibility(
     # Filter out __init__ files
     all_relative_imports = {
         relative_import for relative_import in all_relative_imports
-        if relative_import != '__init__'
+        if relative_import not in {'__init__', 'modeling_mpt'}
     }
     for entrypoint in entrypoint_files:
         file_path = os.path.join(folder, entrypoint)

Original file line number	Diff line number	Diff line change
`@@ -279,7 +279,7 @@ def edit_files_for_hf_compatibility(`
`279`	`279`	`# Filter out __init__ files`
`280`	`280`	`all_relative_imports = {`
`281`	`281`	`relative_import for relative_import in all_relative_imports`
`282`		`- if relative_import != '__init__'`
	`282`	`+ if relative_import not in {'__init__', 'modeling_mpt'}`
`283`	`283`	`}`
`284`	`284`	`for entrypoint in entrypoint_files:`
`285`	`285`	`file_path = os.path.join(folder, entrypoint)`