Hugging Face's logo

Spaces:
Yiyuan
/
VSA
Sleeping

App Files Community
1
cnzzx commited on
Commit
f16f78e
•
1 Parent(s): 7b6b454

update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -100,8 +100,8 @@ with gr.Blocks() as app:
100
 
101
 
102
  vsa = VisionSearchAssistant(
103
- ground_device = "cuda:1",
104
- vlm_device="cuda:1",
105
  vlm_load_4bit=True,
106
  )
107
  # Launch the app
 
100
 
101
 
102
  vsa = VisionSearchAssistant(
103
+ ground_device = "cuda:0",
104
+ vlm_device="cuda:0",
105
  vlm_load_4bit=True,
106
  )
107
  # Launch the app