Update modeling_neollm.py
Browse files- 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
|