gokaygokay
/

moondream-prompt

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

gokaygokay commited on Mar 26

Commit

7f8733b

•

1 Parent(s): ac6c8fc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -18,7 +18,7 @@ from PIL import Image
 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)

 DEVICE = "cuda"
 DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16 # CPU doesn't support float16
+revision = "ac6c8fc0ba757c6c4d7d541fdd0e63618457350c"
 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)