Avoid failure TFBlipModelTest::test_pipeline_image_to_text (#31827)

* fix

* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-07-08 13:49:21 +02:00
committed by GitHub
parent ba743700f4
commit 4879ac2b33
3 changed files with 13 additions and 3 deletions

View File

@@ -17,7 +17,7 @@ import unittest
import requests
from transformers import MODEL_FOR_VISION_2_SEQ_MAPPING, TF_MODEL_FOR_VISION_2_SEQ_MAPPING, is_vision_available
from transformers.pipelines import pipeline
from transformers.pipelines import ImageToTextPipeline, pipeline
from transformers.testing_utils import (
is_pipeline_test,
require_tf,
@@ -46,8 +46,8 @@ class ImageToTextPipelineTests(unittest.TestCase):
tf_model_mapping = TF_MODEL_FOR_VISION_2_SEQ_MAPPING
def get_test_pipeline(self, model, tokenizer, processor, torch_dtype="float32"):
pipe = pipeline(
"image-to-text", model=model, tokenizer=tokenizer, image_processor=processor, torch_dtype=torch_dtype
pipe = ImageToTextPipeline(
model=model, tokenizer=tokenizer, image_processor=processor, torch_dtype=torch_dtype
)
examples = [
Image.open("./tests/fixtures/tests_samples/COCO/000000039769.png"),