John6666 commited on
Commit
c75139c
·
verified ·
1 Parent(s): 29b58ed

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -25,7 +25,6 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
25
  #).to(device)
26
  moondream = ORTModelForImageClassification.from_pretrained(
27
  model_id, trust_remote_code=True,
28
- torch_dtype=torch.bfloat16 if device == "cuda" else torch.float32,
29
  from_transformers=True
30
  ).to(device)
31
 
 
25
  #).to(device)
26
  moondream = ORTModelForImageClassification.from_pretrained(
27
  model_id, trust_remote_code=True,
 
28
  from_transformers=True
29
  ).to(device)
30