shunxing1234
commited on
Commit
•
3f4ba91
1
Parent(s):
98e96d8
Update modeling_aquila.py
Browse files- modeling_aquila.py +3 -3
modeling_aquila.py
CHANGED
@@ -27,9 +27,9 @@ from torch import nn
|
|
27 |
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
|
28 |
|
29 |
from ...activations import ACT2FN
|
30 |
-
from
|
31 |
-
from
|
32 |
-
from
|
33 |
from .configuration_aquila import AquilaConfig
|
34 |
|
35 |
|
|
|
27 |
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
|
28 |
|
29 |
from ...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 .configuration_aquila import AquilaConfig
|
34 |
|
35 |
|