fix import
Browse files
modeling_character_bert.py
CHANGED
@@ -30,15 +30,15 @@ import torch.utils.checkpoint
|
|
30 |
from torch import nn
|
31 |
from torch.nn import CrossEntropyLoss, MSELoss
|
32 |
|
33 |
-
from
|
34 |
-
from
|
35 |
ModelOutput,
|
36 |
add_code_sample_docstrings,
|
37 |
add_start_docstrings,
|
38 |
add_start_docstrings_to_model_forward,
|
39 |
replace_return_docstrings,
|
40 |
)
|
41 |
-
from
|
42 |
BaseModelOutputWithPastAndCrossAttentions,
|
43 |
BaseModelOutputWithPoolingAndCrossAttentions,
|
44 |
CausalLMOutputWithCrossAttentions,
|
@@ -49,13 +49,13 @@ from ...modeling_outputs import (
|
|
49 |
SequenceClassifierOutput,
|
50 |
TokenClassifierOutput,
|
51 |
)
|
52 |
-
from
|
53 |
PreTrainedModel,
|
54 |
apply_chunking_to_forward,
|
55 |
find_pruneable_heads_and_indices,
|
56 |
prune_linear_layer,
|
57 |
)
|
58 |
-
from
|
59 |
from .configuration_character_bert import CharacterBertConfig
|
60 |
from .tokenization_character_bert import CharacterMapper
|
61 |
|
|
|
30 |
from torch import nn
|
31 |
from torch.nn import CrossEntropyLoss, MSELoss
|
32 |
|
33 |
+
from transformers.activations import ACT2FN
|
34 |
+
from transformers.file_utils import (
|
35 |
ModelOutput,
|
36 |
add_code_sample_docstrings,
|
37 |
add_start_docstrings,
|
38 |
add_start_docstrings_to_model_forward,
|
39 |
replace_return_docstrings,
|
40 |
)
|
41 |
+
from transformers.modeling_outputs import (
|
42 |
BaseModelOutputWithPastAndCrossAttentions,
|
43 |
BaseModelOutputWithPoolingAndCrossAttentions,
|
44 |
CausalLMOutputWithCrossAttentions,
|
|
|
49 |
SequenceClassifierOutput,
|
50 |
TokenClassifierOutput,
|
51 |
)
|
52 |
+
from transformers.modeling_utils import (
|
53 |
PreTrainedModel,
|
54 |
apply_chunking_to_forward,
|
55 |
find_pruneable_heads_and_indices,
|
56 |
prune_linear_layer,
|
57 |
)
|
58 |
+
from transformers.utils import logging
|
59 |
from .configuration_character_bert import CharacterBertConfig
|
60 |
from .tokenization_character_bert import CharacterMapper
|
61 |
|