Spaces:
Running
on
Zero
Running
on
Zero
John Ho
commited on
Commit
·
a7fd61f
1
Parent(s):
d9d1598
trying model quantization
Browse files
app.py
CHANGED
|
@@ -92,7 +92,7 @@ def load_model(
|
|
| 92 |
def load_processor(model_name="Qwen/Qwen2.5-VL-7B-Instruct"):
|
| 93 |
return AutoProcessor.from_pretrained(
|
| 94 |
model_name,
|
| 95 |
-
|
| 96 |
use_fast=True,
|
| 97 |
torch_dtype=DTYPE,
|
| 98 |
)
|
|
|
|
| 92 |
def load_processor(model_name="Qwen/Qwen2.5-VL-7B-Instruct"):
|
| 93 |
return AutoProcessor.from_pretrained(
|
| 94 |
model_name,
|
| 95 |
+
device_map=DEVICE,
|
| 96 |
use_fast=True,
|
| 97 |
torch_dtype=DTYPE,
|
| 98 |
)
|