From 2e9af294940083915ccb2740a7c8d5b154194f15 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Wed, 5 Jan 2022 16:58:42 +0100 Subject: [PATCH] [CLIP] Fix TF test (#15042) --- tests/test_modeling_tf_clip.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_modeling_tf_clip.py b/tests/test_modeling_tf_clip.py index 59ef2c49d0..36b78f7c00 100644 --- a/tests/test_modeling_tf_clip.py +++ b/tests/test_modeling_tf_clip.py @@ -254,7 +254,7 @@ class TFCLIPVisionModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: - model = TFCLIPVisionModel.from_pretrained(model_name, from_pt=True) + model = TFCLIPVisionModel.from_pretrained(model_name) self.assertIsNotNone(model) @@ -359,7 +359,7 @@ class TFCLIPTextModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: - model = TFCLIPTextModel.from_pretrained(model_name, from_pt=True) + model = TFCLIPTextModel.from_pretrained(model_name) self.assertIsNotNone(model) @@ -618,7 +618,7 @@ class TFCLIPModelTest(TFModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): for model_name in TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: - model = TFCLIPModel.from_pretrained(model_name, from_pt=True) + model = TFCLIPModel.from_pretrained(model_name) self.assertIsNotNone(model) @@ -630,11 +630,12 @@ def prepare_img(): @require_vision +@require_tf class TFCLIPModelIntegrationTest(unittest.TestCase): @slow def test_inference(self): model_name = "openai/clip-vit-base-patch32" - model = TFCLIPModel.from_pretrained(model_name, from_pt=True) + model = TFCLIPModel.from_pretrained(model_name) processor = CLIPProcessor.from_pretrained(model_name) image = prepare_img()