vikhyatk commited on
Commit
63545fc
1 Parent(s): fda8dae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,9 +14,9 @@ else:
14
  device, dtype = "cpu", torch.float32
15
 
16
  model_id = "vikhyatk/moondream2"
17
- tokenizer = AutoTokenizer.from_pretrained(model_id, revision="2024-03-04")
18
  moondream = AutoModelForCausalLM.from_pretrained(
19
- model_id, trust_remote_code=True, revision="2024-03-04"
20
  ).to(device=device, dtype=dtype)
21
  moondream.eval()
22
 
 
14
  device, dtype = "cpu", torch.float32
15
 
16
  model_id = "vikhyatk/moondream2"
17
+ tokenizer = AutoTokenizer.from_pretrained(model_id, revision="2024-03-06")
18
  moondream = AutoModelForCausalLM.from_pretrained(
19
+ model_id, trust_remote_code=True, revision="2024-03-06"
20
  ).to(device=device, dtype=dtype)
21
  moondream.eval()
22