fix image2test args forwarding (#19648)
* fix image2test args forwarding * fix issues * Proposing the update to the PR. * Fixup. Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
This commit is contained in:
@@ -86,6 +86,12 @@ class ImageToTextPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseMeta
|
||||
],
|
||||
)
|
||||
|
||||
outputs = pipe(image, max_new_tokens=1)
|
||||
self.assertEqual(
|
||||
outputs,
|
||||
[{"generated_text": "growth"}],
|
||||
)
|
||||
|
||||
@require_torch
|
||||
def test_small_model_pt(self):
|
||||
pipe = pipeline("image-to-text", model="hf-internal-testing/tiny-random-vit-gpt2")
|
||||
|
||||
Reference in New Issue
Block a user