Spaces:

vikhyatk
/

moondream2

Running on Zero

vikhyatk commited on Mar 13

Commit

12e7969

•

1 Parent(s): b3229b9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,9 +14,10 @@ else:
     device, dtype = "cpu", torch.float32
 model_id = "vikhyatk/moondream2"
-tokenizer = AutoTokenizer.from_pretrained(model_id, revision="2024-03-13")
 moondream = AutoModelForCausalLM.from_pretrained(
-    model_id, trust_remote_code=True, revision="2024-03-13"
 ).to(device=device, dtype=dtype)
 moondream.eval()

     device, dtype = "cpu", torch.float32
 model_id = "vikhyatk/moondream2"
+revision = "2b705eea63f9bff6dae9b52c2daeb26bc10e4aeb"
+tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
 moondream = AutoModelForCausalLM.from_pretrained(
+    model_id, trust_remote_code=True, revision=revision
 ).to(device=device, dtype=dtype)
 moondream.eval()