gokaygokay
commited on
Commit
•
7f8733b
1
Parent(s):
ac6c8fc
Update README.md
Browse files
README.md
CHANGED
@@ -18,7 +18,7 @@ from PIL import Image
|
|
18 |
|
19 |
DEVICE = "cuda"
|
20 |
DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16 # CPU doesn't support float16
|
21 |
-
revision = "
|
22 |
tokenizer = AutoTokenizer.from_pretrained("gokaygokay/moondream-prompt", revision=revision)
|
23 |
moondream = AutoModelForCausalLM.from_pretrained("gokaygokay/moondream-prompt",trust_remote_code=True,
|
24 |
torch_dtype=DTYPE, device_map={"": DEVICE}, revision=revision)
|
|
|
18 |
|
19 |
DEVICE = "cuda"
|
20 |
DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16 # CPU doesn't support float16
|
21 |
+
revision = "ac6c8fc0ba757c6c4d7d541fdd0e63618457350c"
|
22 |
tokenizer = AutoTokenizer.from_pretrained("gokaygokay/moondream-prompt", revision=revision)
|
23 |
moondream = AutoModelForCausalLM.from_pretrained("gokaygokay/moondream-prompt",trust_remote_code=True,
|
24 |
torch_dtype=DTYPE, device_map={"": DEVICE}, revision=revision)
|