Removal of deprecated maps (#30576)
* [test_all] Remove all imports Remove remaining ARCHIVE MAPS Remove remaining PRETRAINED maps * review comments * [test_all] empty commit to trigger tests
This commit is contained in:
@@ -36,7 +36,6 @@ if is_tf_available():
|
||||
|
||||
from transformers import TFSwiftFormerForImageClassification, TFSwiftFormerModel
|
||||
from transformers.modeling_tf_utils import keras
|
||||
from transformers.models.swiftformer.modeling_tf_swiftformer import TF_SWIFTFORMER_PRETRAINED_MODEL_ARCHIVE_LIST
|
||||
|
||||
|
||||
if is_vision_available():
|
||||
@@ -144,6 +143,7 @@ class TFSwiftFormerModelTest(TFModelTesterMixin, PipelineTesterMixin, unittest.T
|
||||
test_head_masking = False
|
||||
has_attentions = False
|
||||
test_onnx = False
|
||||
from_pretrained_id = "MBZUAI/swiftformer-xs"
|
||||
|
||||
def setUp(self):
|
||||
self.model_tester = TFSwiftFormerModelTester(self)
|
||||
@@ -194,9 +194,8 @@ class TFSwiftFormerModelTest(TFModelTesterMixin, PipelineTesterMixin, unittest.T
|
||||
|
||||
@slow
|
||||
def test_model_from_pretrained(self):
|
||||
for model_name in TF_SWIFTFORMER_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
|
||||
model = TFSwiftFormerModel.from_pretrained(model_name)
|
||||
self.assertIsNotNone(model)
|
||||
model = TFSwiftFormerModel.from_pretrained(self.from_pretrained_id)
|
||||
self.assertIsNotNone(model)
|
||||
|
||||
@unittest.skip(reason="TFSwiftFormer does not output attentions")
|
||||
def test_attention_outputs(self):
|
||||
|
||||
Reference in New Issue
Block a user