Make more test models smaller (#25005)
* Make more test models tiny * Make more test models tiny * More models * More models
This commit is contained in:
@@ -56,7 +56,7 @@ class MobileViTModelTester:
|
||||
image_size=32,
|
||||
patch_size=2,
|
||||
num_channels=3,
|
||||
last_hidden_size=640,
|
||||
last_hidden_size=32,
|
||||
num_attention_heads=4,
|
||||
hidden_act="silu",
|
||||
conv_kernel_size=3,
|
||||
@@ -115,6 +115,8 @@ class MobileViTModelTester:
|
||||
attention_probs_dropout_prob=self.attention_probs_dropout_prob,
|
||||
classifier_dropout_prob=self.classifier_dropout_prob,
|
||||
initializer_range=self.initializer_range,
|
||||
hidden_sizes=[12, 16, 20],
|
||||
neck_hidden_sizes=[8, 8, 16, 16, 32, 32, 32],
|
||||
)
|
||||
|
||||
def create_and_check_model(self, config, pixel_values, labels, pixel_labels):
|
||||
@@ -231,10 +233,6 @@ class MobileViTModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
|
||||
expected_arg_names = ["pixel_values"]
|
||||
self.assertListEqual(arg_names[:1], expected_arg_names)
|
||||
|
||||
@unittest.skip("Will be fixed soon by reducing the size of the model used for common tests.")
|
||||
def test_model_is_small(self):
|
||||
pass
|
||||
|
||||
def test_model(self):
|
||||
config_and_inputs = self.model_tester.prepare_config_and_inputs()
|
||||
self.model_tester.create_and_check_model(*config_and_inputs)
|
||||
|
||||
Reference in New Issue
Block a user