Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
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 |
|