it's Friday night, let cross finger

huggingface · Dec 13, 2024 · 00e324d · 00e324d
1 parent 90c4fcc
commit 00e324d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/kosmos2_5/test_processor_kosmos2_5.py b/tests/models/kosmos2_5/test_processor_kosmos2_5.py
@@ -47,6 +47,7 @@
 @require_vision
 class Kosmos2_5ProcessorTest(ProcessorTesterMixin, unittest.TestCase):
     processor_class = Kosmos2_5Processor
+    images_input_name = "flattened_patches"
 
     def setUp(self):
         self.tmpdirname = tempfile.mkdtemp()