Return BaseModelOutputWithPoolingAndCrossAttentions
Browse files- modeling_flexbert.py +1 -1
modeling_flexbert.py
CHANGED
|
@@ -66,7 +66,7 @@ from transformers.modeling_outputs import (
|
|
| 66 |
SequenceClassifierOutput,
|
| 67 |
)
|
| 68 |
from transformers.models.bert.modeling_bert import BertPreTrainedModel
|
| 69 |
-
from transformers import BaseModelOutputWithPoolingAndCrossAttentions
|
| 70 |
from .bert_padding import index_put_first_axis
|
| 71 |
|
| 72 |
from .activation import get_act_fn
|
|
|
|
| 66 |
SequenceClassifierOutput,
|
| 67 |
)
|
| 68 |
from transformers.models.bert.modeling_bert import BertPreTrainedModel
|
| 69 |
+
from transformers.modeling_outputs import BaseModelOutputWithPoolingAndCrossAttentions
|
| 70 |
from .bert_padding import index_put_first_axis
|
| 71 |
|
| 72 |
from .activation import get_act_fn
|