Avoid failure TFBlipModelTest::test_pipeline_image_to_text (#31827)

* fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-07-08 13:49:21 +02:00
parent ba743700f4
commit 4879ac2b33
3 changed files with 13 additions and 3 deletions
--- a/tests/models/efficientnet/test_modeling_efficientnet.py
+++ b/tests/models/efficientnet/test_modeling_efficientnet.py
@@ -215,6 +215,12 @@ class EfficientNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Test
    def test_pipeline_image_feature_extraction(self):
        super().test_pipeline_image_feature_extraction()

+    @is_pipeline_test
+    @require_vision
+    @slow
+    def test_pipeline_image_feature_extraction_fp16(self):
+        super().test_pipeline_image_feature_extraction_fp16()
+
    @is_pipeline_test
    @require_vision
    @slow