Update README.md
Browse files
README.md
CHANGED
@@ -89,7 +89,7 @@ image3 = load_image("https://cdn.britannica.com/68/170868-050-8DDE8263/Golden-Ga
|
|
89 |
|
90 |
processor = AutoProcessor.from_pretrained("HuggingFaceM4/Idefics3-8B-Llama3")
|
91 |
model = AutoModelForVision2Seq.from_pretrained(
|
92 |
-
"HuggingFaceM4/Idefics3-8B-Llama3",
|
93 |
).to(DEVICE)
|
94 |
|
95 |
# Create inputs
|
|
|
89 |
|
90 |
processor = AutoProcessor.from_pretrained("HuggingFaceM4/Idefics3-8B-Llama3")
|
91 |
model = AutoModelForVision2Seq.from_pretrained(
|
92 |
+
"HuggingFaceM4/Idefics3-8B-Llama3", torch_dtype=torch.bfloat16
|
93 |
).to(DEVICE)
|
94 |
|
95 |
# Create inputs
|