cnzzx commited on
Commit
2bfc524
1 Parent(s): 7551acb
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -100,9 +100,9 @@ with gr.Blocks() as app:
100
 
101
 
102
  vsa = VisionSearchAssistant(
103
- ground_device = "cuda:1",
104
- vlm_device="cuda:2",
105
- vlm_load_4bit=True,
106
  )
107
  # Launch the app
108
  app.launch()
 
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
108
  app.launch()