KitsuVp commited on
Commit
1e1d6b7
·
verified ·
1 Parent(s): e2d023d

Update modeling_neollm.py

Browse files
Files changed (1) hide show
  1. modeling_neollm.py +1 -1
modeling_neollm.py CHANGED
@@ -60,7 +60,7 @@ from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_u
60
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
61
  from transformers.processing_utils import Unpack
62
  from transformers.utils import TransformersKwargs, logging
63
- from configuration_neollm import NeoLLMConfig
64
 
65
  from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
66
 
 
60
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
61
  from transformers.processing_utils import Unpack
62
  from transformers.utils import TransformersKwargs, logging
63
+ from .configuration_neollm import NeoLLMConfig
64
 
65
  from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
66