add fcs back

mosaicml · May 27, 2024 · 1507c8b · 1507c8b
1 parent d80087f
commit 1507c8b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/llmfoundry/models/mpt/modeling_mpt.py b/llmfoundry/models/mpt/modeling_mpt.py
@@ -75,6 +75,12 @@
     mpt_get_total_params,
 )
 
+# Import the fcs here so that recursive code creating the files for hf checkpoints can find them
+# This is the only exception because fc.py is not imported in any other place in the codebase
+# isort: off
+from llmfoundry.models.layers.fc import fcs  #  type: ignore
+# isort: on
+
 log = logging.getLogger(__name__)