Skip to content

Commit 3fbf54b

Browse files
committed
Fix tutorial error
1 parent c1b1f44 commit 3fbf54b

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

docs/LLaVA_OneVision_Tutorials.ipynb

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,13 @@
176176
"model_name = \"llava_qwen\"\n",
177177
"device = \"cuda\"\n",
178178
"device_map = \"auto\"\n",
179-
"tokenizer, model, image_processor, max_length = load_pretrained_model(pretrained, None, model_name, device_map=device_map)\n",
179+
"llava_model_args = {\n",
180+
" \"multimodal\": True,\n",
181+
" }\n",
182+
"overwrite_config = {}\n",
183+
"overwrite_config[\"image_aspect_ratio\"] = \"pad\"\n",
184+
"llava_model_args[\"overwrite_config\"] = overwrite_config\n",
185+
"tokenizer, model, image_processor, max_length = load_pretrained_model(pretrained, None, model_name, device_map=device_map, **llava_model_args)\n",
180186
"\n",
181187
"model.eval()\n",
182188
"\n",
@@ -299,6 +305,7 @@
299305
" do_sample=False,\n",
300306
" temperature=0,\n",
301307
" max_new_tokens=4096,\n",
308+
" modalities=[\"video\"],\n",
302309
")\n",
303310
"text_outputs = tokenizer.batch_decode(cont, skip_special_tokens=True)\n",
304311
"print(text_outputs[0])"
@@ -307,7 +314,7 @@
307314
],
308315
"metadata": {
309316
"kernelspec": {
310-
"display_name": "llava",
317+
"display_name": "Python 3.9.2 64-bit",
311318
"language": "python",
312319
"name": "python3"
313320
},
@@ -321,7 +328,12 @@
321328
"name": "python",
322329
"nbconvert_exporter": "python",
323330
"pygments_lexer": "ipython3",
324-
"version": "3.10.14"
331+
"version": "3.9.2"
332+
},
333+
"vscode": {
334+
"interpreter": {
335+
"hash": "31f2aee4e71d21fbe5cf8b01ff0e069b9275f58929596ceb00d14d90e3e16cd6"
336+
}
325337
}
326338
},
327339
"nbformat": 4,

0 commit comments

Comments
 (0)