update images input name in nanollava notebook (#2529)

openvinotoolkit · Nov 18, 2024 · ac90d38 · ac90d38
1 parent 9868edb
commit ac90d38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/nano-llava-multimodal-chatbot/nano-llava-multimodal-chatbot.ipynb b/notebooks/nano-llava-multimodal-chatbot/nano-llava-multimodal-chatbot.ipynb
@@ -672,7 +672,7 @@
     "print(f\"Question:\\n{prompt}\")\n",
     "print(\"Answer:\")\n",
     "\n",
-    "output_ids = ov_model.generate(input_ids, attention_mask=attention_mask, images=image_tensor, max_new_tokens=128, use_cache=True, streamer=streamer)"
+    "output_ids = ov_model.generate(input_ids, attention_mask=attention_mask, pixel_values=image_tensor, max_new_tokens=128, use_cache=True, streamer=streamer)"
    ]
   },
   {
@@ -774,7 +774,7 @@
     "    generation_kwargs = dict(\n",
     "        input_ids=input_ids,\n",
     "        attention_mask=attention_mask,\n",
-    "        images=image_tensor,\n",
+    "        pixel_values=image_tensor,\n",
     "        streamer=streamer,\n",
     "        max_new_tokens=128,\n",
     "        stopping_criteria=[stopping_criteria],\n",