gokaygokay commited on
Commit
7f8733b
1 Parent(s): ac6c8fc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = "1082928d0aa39290a31a92f632ca670458eda512"
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)