Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -72,7 +72,7 @@ try:
|
|
| 72 |
MODEL_PATH,
|
| 73 |
torch_dtype=torch.bfloat16 if DEVICE == "cuda" else torch.float32,
|
| 74 |
device_map="auto" if DEVICE == "cuda" else None
|
| 75 |
-
)
|
| 76 |
processor = AutoProcessor.from_pretrained(MODEL_PATH, use_fast=True)
|
| 77 |
print(f"Qwen model loaded on {DEVICE}")
|
| 78 |
except Exception as e:
|
|
|
|
| 72 |
MODEL_PATH,
|
| 73 |
torch_dtype=torch.bfloat16 if DEVICE == "cuda" else torch.float32,
|
| 74 |
device_map="auto" if DEVICE == "cuda" else None
|
| 75 |
+
)
|
| 76 |
processor = AutoProcessor.from_pretrained(MODEL_PATH, use_fast=True)
|
| 77 |
print(f"Qwen model loaded on {DEVICE}")
|
| 78 |
except Exception as e:
|