layer.py -> block.py

2 years ago · ffce7718ac
parent 9c929b6f3b
commit ffce7718ac
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.bloom_block import BloomBlock
+from src.block import BloomBlock
 from src.ops import build_alibi_tensor

 logger = logging.get_logger(__name__)