Hugging Face's logo

Spaces:
AI-Safeguard
/
Ivy-VL
Running

App Files Community
Ivy1997 commited on
Commit
7acf621
·
verified ·
1 Parent(s): 7ac54c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ warnings.filterwarnings("ignore")
19
 
20
  pretrained = "AI-Safeguard/Ivy-VL-llava"
21
  model_name = "llava_qwen"
22
- device = "cuda"
23
  device_map = "auto"
24
 
25
  # Load model, tokenizer, and image processor
 
19
 
20
  pretrained = "AI-Safeguard/Ivy-VL-llava"
21
  model_name = "llava_qwen"
22
+ device = "cpu"
23
  device_map = "auto"
24
 
25
  # Load model, tokenizer, and image processor