pass text and images as kwargs to VLM processor

lapp0 · Aug 30, 2024 · 18eafb0 · 18eafb0
1 parent 7cf63d6
commit 18eafb0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/outlines/models/transformers_vision.py b/outlines/models/transformers_vision.py
@@ -43,9 +43,9 @@ def generate(  # type: ignore
         -------
         The generated text
         """
-        inputs = self.processor(prompts, media, padding=True, return_tensors="pt").to(
-            self.model.device
-        )
+        inputs = self.processor(
+            text=prompts, images=media, padding=True, return_tensors="pt"
+        ).to(self.model.device)
 
         generation_kwargs = self._get_generation_kwargs(
             prompts,