Stremly commited on
Commit
7de7341
·
verified ·
1 Parent(s): 3a49137

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from qwen_vl_utils import process_vision_info # include this file in your repo
12
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
13
  "ByteDance-Seed/UI-TARS-1.5-7B",
14
  device_map="auto",
15
- torch_dtype=torch.float32, # CPU‑friendly
16
  )
17
  processor = AutoProcessor.from_pretrained(
18
  "ByteDance-Seed/UI-TARS-1.5-7B",
 
12
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
13
  "ByteDance-Seed/UI-TARS-1.5-7B",
14
  device_map="auto",
15
+ torch_dtype=torch.float16, # CPU‑friendly
16
  )
17
  processor = AutoProcessor.from_pretrained(
18
  "ByteDance-Seed/UI-TARS-1.5-7B",