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:
Sylvain Gugger
2023-07-25 07:56:04 -04:00
committed by GitHub
parent ee1eb3b325
commit f295fc8a16
19 changed files with 109 additions and 108 deletions

View File

@@ -42,7 +42,7 @@ class TimmBackboneModelTester:
out_indices=None,
out_features=None,
stage_names=None,
backbone="resnet50",
backbone="resnet18",
batch_size=3,
image_size=32,
num_channels=3,
@@ -196,7 +196,7 @@ class TimmBackboneModelTest(ModelTesterMixin, BackboneTesterMixin, PipelineTeste
def test_can_use_safetensors(self):
pass
@unittest.skip("Will be fixed soon by reducing the size of the model used for common tests.")
@unittest.skip("Need to use a timm backbone and there is no tiny model available.")
def test_model_is_small(self):
pass