KitsuVp commited on
Commit
bea8374
·
verified ·
1 Parent(s): cb9da8b

Update modeling_neollm.py

Browse files
Files changed (1) hide show
  1. modeling_neollm.py +1 -1
modeling_neollm.py CHANGED
@@ -88,7 +88,7 @@ from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_u
88
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
89
  from transformers.processing_utils import Unpack
90
  from transformers.utils import TransformersKwargs, logging
91
- from configuration_neollm import NeoLLMConfig
92
 
93
  from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
94
  torch._dynamo.config.capture_scalar_outputs = True
 
88
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
89
  from transformers.processing_utils import Unpack
90
  from transformers.utils import TransformersKwargs, logging
91
+ from .configuration_neollm import NeoLLMConfig
92
 
93
  from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
94
  torch._dynamo.config.capture_scalar_outputs = True