Skip to content

Commit ac90d38

Browse files
authored
update images input name in nanollava notebook (#2529)
1 parent 9868edb commit ac90d38

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

notebooks/nano-llava-multimodal-chatbot/nano-llava-multimodal-chatbot.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -672,7 +672,7 @@
672672
"print(f\"Question:\\n{prompt}\")\n",
673673
"print(\"Answer:\")\n",
674674
"\n",
675-
"output_ids = ov_model.generate(input_ids, attention_mask=attention_mask, images=image_tensor, max_new_tokens=128, use_cache=True, streamer=streamer)"
675+
"output_ids = ov_model.generate(input_ids, attention_mask=attention_mask, pixel_values=image_tensor, max_new_tokens=128, use_cache=True, streamer=streamer)"
676676
]
677677
},
678678
{
@@ -774,7 +774,7 @@
774774
" generation_kwargs = dict(\n",
775775
" input_ids=input_ids,\n",
776776
" attention_mask=attention_mask,\n",
777-
" images=image_tensor,\n",
777+
" pixel_values=image_tensor,\n",
778778
" streamer=streamer,\n",
779779
" max_new_tokens=128,\n",
780780
" stopping_criteria=[stopping_criteria],\n",

0 commit comments

Comments
 (0)