Patch-past-refactor (#21050)

* small patches, forgot a line

* refactor PT

* the actual fix
This commit is contained in:
Arthur
2023-01-09 18:12:13 +01:00
committed by GitHub
parent 48d4e147d8
commit e3ecbaa4ab
4 changed files with 5 additions and 8 deletions

View File

@@ -55,7 +55,6 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta
)
@require_tf
@unittest.skip("Arthur will fix me!")
def test_small_model_tf(self):
pipe = pipeline("image-to-text", model="hf-internal-testing/tiny-random-vit-gpt2", framework="tf")
image = "./tests/fixtures/tests_samples/COCO/000000039769.png"