oweller2 commited on
Commit ·
c116d9c
1
Parent(s): f7dca47
lm output
Browse files- modeling_flexbert.py +1 -0
modeling_flexbert.py
CHANGED
|
@@ -62,6 +62,7 @@ from torch.nn.modules.utils import consume_prefix_in_state_dict_if_present
|
|
| 62 |
from transformers.modeling_outputs import (
|
| 63 |
MaskedLMOutput,
|
| 64 |
ModelOutput,
|
|
|
|
| 65 |
MultipleChoiceModelOutput,
|
| 66 |
SequenceClassifierOutput,
|
| 67 |
)
|
|
|
|
| 62 |
from transformers.modeling_outputs import (
|
| 63 |
MaskedLMOutput,
|
| 64 |
ModelOutput,
|
| 65 |
+
CausalLMOutput,
|
| 66 |
MultipleChoiceModelOutput,
|
| 67 |
SequenceClassifierOutput,
|
| 68 |
)
|