layer.py -> bloom_block.py

2 years ago · 9c929b6f3b
parent 79925fc5f7
commit 9c929b6f3b
2 changed files with 1 additions and 1 deletions
--- a/src/bloom_block.py
+++ b/src/bloom_block.py
--- a/src/model.py
+++ b/src/model.py
@ -17,7 +17,7 @@ from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import logging
 from transformers.models.bloom.configuration_bloom import BloomConfig as _VanillaBloomConfig

-from src.layer import BloomBlock
+from src.bloom_block import BloomBlock
 from src.ops import build_alibi_tensor

 logger = logging.get_logger(__name__)