vikhyatk
/

moondream2

Image-Text-to-Text

text-generation

Model card Files Files and versions

vikhyatk commited on Mar 5, 2024

Commit

fc49662

·

verified ·

1 Parent(s): 4a8fa31

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -21,10 +21,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from PIL import Image
 model_id = "vikhyatk/moondream2"
 model = AutoModelForCausalLM.from_pretrained(
-    model_id, trust_remote_code=True, revision="2024-03-04"
 )
-tokenizer = AutoTokenizer.from_pretrained(model_id, revision="2024-03-04")
 image = Image.open('<IMAGE_PATH>')
 enc_image = model.encode_image(image)

 from PIL import Image
 model_id = "vikhyatk/moondream2"
+revision = "2024-03-05"
 model = AutoModelForCausalLM.from_pretrained(
+    model_id, trust_remote_code=True, revision=revision
 )
+tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
 image = Image.open('<IMAGE_PATH>')
 enc_image = model.encode_image(image)