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