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 |
|