Skip to content

Commit aca8d11

Browse files
authored
[ARM] Fix bug in phi-3-vision.ipynb (#2693)
@eaidova
1 parent 93b4c69 commit aca8d11

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

notebooks/phi-3-vision/phi-3-vision.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -455,7 +455,7 @@
455455
" {\"role\": \"user\", \"content\": \"<|image_1|>\\nWhat is unusual on this picture?\"},\n",
456456
"]\n",
457457
"\n",
458-
"processor = AutoProcessor.from_pretrained(MODEL_DIR / \"INT4\", trust_remote_code=True)\n",
458+
"processor = AutoProcessor.from_pretrained(MODEL_DIR / \"INT4\" if to_compress.value else \"FP16\", trust_remote_code=True)\n",
459459
"\n",
460460
"prompt = processor.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)\n",
461461
"\n",

0 commit comments

Comments
 (0)