Spaces:

vikhyatk
/

moondream2

Running on Zero

vikhyatk commited on Mar 13

Commit

b3229b9

•

1 Parent(s): d9ab7d8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     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()