Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -89,7 +89,7 @@ load_error_message = ""
|
|
89 |
try:
|
90 |
model = AutoModelForImageTextToText.from_pretrained(
|
91 |
MODEL_ID,
|
92 |
-
torch_dtype=torch.
|
93 |
trust_remote_code=True
|
94 |
).to(pick_device())
|
95 |
processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
|
|
|
89 |
try:
|
90 |
model = AutoModelForImageTextToText.from_pretrained(
|
91 |
MODEL_ID,
|
92 |
+
torch_dtype=torch.bfloat16, # CPU-friendly
|
93 |
trust_remote_code=True
|
94 |
).to(pick_device())
|
95 |
processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
|