Minthy
/

ToriiGate-v0.2

Image-Text-to-Text

Model card Files Files and versions Community

Minthy commited on Sep 26

Commit

b12dd82

•

1 Parent(s): 361dd2a

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -86,9 +86,9 @@ from transformers import AutoProcessor, Idefics3ForConditionalGeneration
 from transformers.image_utils import load_image
 DEVICE = "cuda:0" #change to your device
-processor = AutoProcessor.from_pretrained("Minthy/Torii_Gate_v0.2") #or change to local path
 model = Idefics3ForConditionalGeneration.from_pretrained(
-    "Minthy/Torii_Gate_v0.2",
 	torch_dtype=torch.bfloat16,
 #	_attn_implementation="flash_attention_2",
 ).to(DEVICE)

 from transformers.image_utils import load_image
 DEVICE = "cuda:0" #change to your device
+processor = AutoProcessor.from_pretrained("Minthy/ToriiGate-v0.2") #or change to local path
 model = Idefics3ForConditionalGeneration.from_pretrained(
+    "Minthy/ToriiGate-v0.2",
 	torch_dtype=torch.bfloat16,
 #	_attn_implementation="flash_attention_2",
 ).to(DEVICE)