Fix test and docs (#14399)

2021-11-15 17:35:33 +01:00
parent 4ce74edf51
commit 74e6111ba7
2 changed files with 3 additions and 2 deletions
--- a/tests/test_modeling_tf_vit.py
+++ b/tests/test_modeling_tf_vit.py
@@ -371,7 +371,7 @@ class TFViTModelIntegrationTest(unittest.TestCase):

    @slow
    def test_inference_image_classification_head(self):
-        model = TFViTForImageClassification.from_pretrained("google/vit-base-patch16-224", from_pt=True)
+        model = TFViTForImageClassification.from_pretrained("google/vit-base-patch16-224")

        feature_extractor = self.default_feature_extractor
        image = prepare_img()