Update README.md
Browse files
README.md
CHANGED
|
@@ -86,9 +86,9 @@ from transformers import AutoProcessor, Idefics3ForConditionalGeneration
|
|
| 86 |
from transformers.image_utils import load_image
|
| 87 |
|
| 88 |
DEVICE = "cuda:0" #change to your device
|
| 89 |
-
processor = AutoProcessor.from_pretrained("Minthy/
|
| 90 |
model = Idefics3ForConditionalGeneration.from_pretrained(
|
| 91 |
-
"Minthy/
|
| 92 |
torch_dtype=torch.bfloat16,
|
| 93 |
# _attn_implementation="flash_attention_2",
|
| 94 |
).to(DEVICE)
|
|
|
|
| 86 |
from transformers.image_utils import load_image
|
| 87 |
|
| 88 |
DEVICE = "cuda:0" #change to your device
|
| 89 |
+
processor = AutoProcessor.from_pretrained("Minthy/ToriiGate-v0.2") #or change to local path
|
| 90 |
model = Idefics3ForConditionalGeneration.from_pretrained(
|
| 91 |
+
"Minthy/ToriiGate-v0.2",
|
| 92 |
torch_dtype=torch.bfloat16,
|
| 93 |
# _attn_implementation="flash_attention_2",
|
| 94 |
).to(DEVICE)
|