Xin Wang
commited on
Commit
•
dbb2e79
1
Parent(s):
b0a5bc7
WIP
Browse files- modeling_gpt2.py +2 -2
modeling_gpt2.py
CHANGED
|
@@ -35,7 +35,7 @@ from transformers.modeling_outputs import (
|
|
| 35 |
SequenceClassifierOutputWithPast,
|
| 36 |
TokenClassifierOutput,
|
| 37 |
)
|
| 38 |
-
from transformers.modeling_utils import SequenceSummary
|
| 39 |
from transformers.pytorch_utils import (
|
| 40 |
Conv1D,
|
| 41 |
find_pruneable_heads_and_indices,
|
|
@@ -50,7 +50,7 @@ from transformers.utils import (
|
|
| 50 |
replace_return_docstrings,
|
| 51 |
)
|
| 52 |
from .configuration_gpt2 import GPT2Config
|
| 53 |
-
from mltools.dmx import DmxPreTrainedModel as PreTrainedModel
|
| 54 |
|
| 55 |
logger = logging.get_logger(__name__)
|
| 56 |
|
|
|
|
| 35 |
SequenceClassifierOutputWithPast,
|
| 36 |
TokenClassifierOutput,
|
| 37 |
)
|
| 38 |
+
from transformers.modeling_utils import PreTrainedModel, SequenceSummary
|
| 39 |
from transformers.pytorch_utils import (
|
| 40 |
Conv1D,
|
| 41 |
find_pruneable_heads_and_indices,
|
|
|
|
| 50 |
replace_return_docstrings,
|
| 51 |
)
|
| 52 |
from .configuration_gpt2 import GPT2Config
|
| 53 |
+
# from mltools.dmx import DmxPreTrainedModel as PreTrainedModel
|
| 54 |
|
| 55 |
logger = logging.get_logger(__name__)
|
| 56 |
|