John Ho commited on
Commit
a7fd61f
·
1 Parent(s): d9d1598

trying model quantization

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- # device_map=DEVICE,
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
  )