Skip to content

Commit a3427d7

Browse files
committed
Fix CI failures
1 parent bd09746 commit a3427d7

File tree

3 files changed

+3
-9
lines changed

3 files changed

+3
-9
lines changed

src/transformers/__init__.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -822,10 +822,7 @@
822822
"TapasConfig",
823823
"TapasTokenizer",
824824
],
825-
"models.textnet": [
826-
"TEXTNET_PRETRAINED_CONFIG_ARCHIVE_MAP",
827-
"TextNetConfig"
828-
],
825+
"models.textnet": ["TEXTNET_PRETRAINED_CONFIG_ARCHIVE_MAP", "TextNetConfig"],
829826
"models.time_series_transformer": [
830827
"TIME_SERIES_TRANSFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP",
831828
"TimeSeriesTransformerConfig",

src/transformers/models/auto/modeling_auto.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -95,10 +95,7 @@
9595
("ernie_m", "ErnieMModel"),
9696
("esm", "EsmModel"),
9797
("falcon", "FalconModel"),
98-
<<<<<<< HEAD
9998
("fastspeech2_conformer", "FastSpeech2ConformerModel"),
100-
=======
101-
>>>>>>> ae576e088 (Remove all the references of fast model)
10299
("flaubert", "FlaubertModel"),
103100
("flava", "FlavaModel"),
104101
("fnet", "FNetModel"),

tests/models/textnet/test_modeling_textnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def prepare_config_and_inputs_for_common(self):
197197
return config, inputs_dict
198198

199199

200-
# Copied from tests.test_modeling_bit.BitBackboneTest with Bit -> TextNet
200+
# Copied from tests.models.bit.test_modeling_bit.BitBackboneTest with Bit -> TextNet
201201
@require_torch
202202
class TextNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
203203
all_model_classes = (TextNetModel, TextNetForImageClassification, TextNetBackbone) if is_torch_available() else ()
@@ -327,7 +327,7 @@ def test_inference_textnet_image_classification(self):
327327
self.assertEqual(output.logits.shape, torch.Size([1, 2]))
328328

329329

330-
# Copied from tests.test_modeling_bit.BitBackboneTest with Bit -> TextNet
330+
# Copied from tests.models.bit.test_modeling_bit.BitBackboneTest with Bit -> TextNet
331331
@require_torch
332332
class TextNetBackboneTest(BackboneTesterMixin, unittest.TestCase):
333333
all_model_classes = (TextNetBackbone,) if is_torch_available() else ()

0 commit comments

Comments
 (0)