Update modeling_neollm.py
Browse files- modeling_neollm.py +1 -1
modeling_neollm.py
CHANGED
|
@@ -71,7 +71,7 @@ from transformers.modeling_rope_utils import ROPE_INIT_FUNCTIONS, dynamic_rope_u
|
|
| 71 |
from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
|
| 72 |
from transformers.processing_utils import Unpack
|
| 73 |
from transformers.utils import TransformersKwargs, logging
|
| 74 |
-
from configuration_neollm import NeoLLMConfig
|
| 75 |
|
| 76 |
from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
|
| 77 |
|
|
|
|
| 71 |
from transformers.modeling_utils import ALL_ATTENTION_FUNCTIONS, PreTrainedModel
|
| 72 |
from transformers.processing_utils import Unpack
|
| 73 |
from transformers.utils import TransformersKwargs, logging
|
| 74 |
+
from .configuration_neollm import NeoLLMConfig
|
| 75 |
|
| 76 |
from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
|
| 77 |
|