gokaygokay
/

moondream-prompt

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

gokaygokay commited on Mar 26

Commit

f741f12

•

1 Parent(s): 1082928

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -18,10 +18,10 @@ from PIL import Image
 DEVICE = "cuda"
 DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16 # CPU doesn't support float16
-tokenizer = AutoTokenizer.from_pretrained("gokaygokay/moondream-prompt")
 moondream = AutoModelForCausalLM.from_pretrained("gokaygokay/moondream-prompt",trust_remote_code=True,
-    torch_dtype=DTYPE, device_map={"": DEVICE})
 moondream.eval()
 image_path = "<image_path>"

 DEVICE = "cuda"
 DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16 # CPU doesn't support float16
+revision = "1082928d0aa39290a31a92f632ca670458eda512"
+tokenizer = AutoTokenizer.from_pretrained("gokaygokay/moondream-prompt", revision=revision)
 moondream = AutoModelForCausalLM.from_pretrained("gokaygokay/moondream-prompt",trust_remote_code=True,
+    torch_dtype=DTYPE, device_map={"": DEVICE}, revision=revision)
 moondream.eval()
 image_path = "<image_path>"