Update modeling_aquila.py
Browse files- modeling_aquila.py +1 -3
modeling_aquila.py
CHANGED
@@ -30,7 +30,7 @@ from transformers.activations import ACT2FN
|
|
30 |
from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, SequenceClassifierOutputWithPast
|
31 |
from transformers.modeling_utils import PreTrainedModel
|
32 |
from transformers.utils import add_start_docstrings, add_start_docstrings_to_model_forward, logging, replace_return_docstrings
|
33 |
-
from
|
34 |
from transformers import (
|
35 |
LogitsProcessorList,
|
36 |
MinLengthLogitsProcessor,
|
@@ -41,8 +41,6 @@ from transformers import (
|
|
41 |
MaxLengthCriteria,
|
42 |
BitsAndBytesConfig,
|
43 |
)
|
44 |
-
from .utils import *
|
45 |
-
|
46 |
|
47 |
logger = logging.get_logger(__name__)
|
48 |
|
|
|
30 |
from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast, SequenceClassifierOutputWithPast
|
31 |
from transformers.modeling_utils import PreTrainedModel
|
32 |
from transformers.utils import add_start_docstrings, add_start_docstrings_to_model_forward, logging, replace_return_docstrings
|
33 |
+
from configuration_aquila import AquilaConfig
|
34 |
from transformers import (
|
35 |
LogitsProcessorList,
|
36 |
MinLengthLogitsProcessor,
|
|
|
41 |
MaxLengthCriteria,
|
42 |
BitsAndBytesConfig,
|
43 |
)
|
|
|
|
|
44 |
|
45 |
logger = logging.get_logger(__name__)
|
46 |
|