KitsuVp commited on
Commit
3542034
·
verified ·
1 Parent(s): 6a59e0a

Update modeling_neollm.py

Browse files
Files changed (1) hide show
  1. modeling_neollm.py +1 -1
modeling_neollm.py CHANGED
@@ -77,7 +77,7 @@ from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_u
77
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
78
  from transformers.processing_utils import Unpack
79
  from transformers.utils import TransformersKwargs, logging
80
- from configuration_neollm import NeoLLMConfig
81
 
82
  from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
83
  torch._dynamo.config.capture_scalar_outputs = True
 
77
  from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
78
  from transformers.processing_utils import Unpack
79
  from transformers.utils import TransformersKwargs, logging
80
+ from .configuration_neollm import NeoLLMConfig
81
 
82
  from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
83
  torch._dynamo.config.capture_scalar_outputs = True