pc

mosaicml · Apr 11, 2024 · 8954a05 · 8954a05
1 parent 2006b92
commit 8954a05
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/llmfoundry/models/utils/act_ckpt.py b/llmfoundry/models/utils/act_ckpt.py
@@ -6,7 +6,6 @@
 import torch
 
 from llmfoundry.layers_registry import attention_classes, norms
-from llmfoundry.models.layers.blocks import MPTBlock
 from llmfoundry.models.layers.blocks import FusedNormAttentionNorm, MPTBlock
 from llmfoundry.models.layers.ffn import FFN_CLASS_REGISTRY