Fix last models for common tests that are too big. (#25058)
* Fix last models for common tests that are too big. * Remove print statement
This commit is contained in:
@@ -58,9 +58,9 @@ class SwiftFormerModelTester:
|
||||
hidden_dropout_prob=0.1,
|
||||
attention_probs_dropout_prob=0.1,
|
||||
image_size=224,
|
||||
num_labels=1000,
|
||||
layer_depths=[3, 3, 6, 4],
|
||||
embed_dims=[48, 56, 112, 220],
|
||||
num_labels=3,
|
||||
layer_depths=[1, 1, 1, 1],
|
||||
embed_dims=[16, 16, 32, 32],
|
||||
):
|
||||
self.parent = parent
|
||||
self.batch_size = batch_size
|
||||
@@ -272,10 +272,6 @@ class SwiftFormerModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestC
|
||||
msg=f"Parameter {name} of model {model_class} seems not properly initialized",
|
||||
)
|
||||
|
||||
@unittest.skip("Will be fixed soon by reducing the size of the model used for common tests.")
|
||||
def test_model_is_small(self):
|
||||
pass
|
||||
|
||||
|
||||
# We will verify our results on an image of cute cats
|
||||
def prepare_img():
|
||||
|
||||
Reference in New Issue
Block a user