Fix merge

mosaicml · Feb 6, 2024 · 82434b2 · 82434b2
1 parent d0cbd11
commit 82434b2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llmfoundry/models/mpt/configuration_mpt.py b/llmfoundry/models/mpt/configuration_mpt.py
@@ -19,9 +19,9 @@
 # isort: off
 from llmfoundry.models.layers.fc import FC_CLASS_REGISTRY  # type: ignore (see note)
 from llmfoundry.models.layers.norm import LPLayerNorm  # type: ignore (see note)
-from llmfoundry.models.layers.ffn import FFN_CLASS_REGISTRY
+from llmfoundry.models.layers.ffn import FFN_CLASS_REGISTRY  # type: ignore (see note)
 
-from llmfoundry.utils.warnings import VersionedDeprecationWarning  # type: ignore (see note)
+from llmfoundry.utils.warnings import VersionedDeprecationWarning
 
 ffn_config_defaults: Dict = {
     'ffn_type': 'mptmlp',