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 |
)
|