Spaces:

Yiyuan
/

VSA

Runtime error

cnzzx commited on Oct 29, 2024

Commit

c3f7520

1 Parent(s): 2bfc524

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,7 +102,7 @@ with gr.Blocks() as app:
 vsa = VisionSearchAssistant(
     ground_device = "cuda:0",
     vlm_device="cuda:0",
-    # vlm_load_4bit=True,
 )
 # Launch the app
 app.launch()

 vsa = VisionSearchAssistant(
     ground_device = "cuda:0",
     vlm_device="cuda:0",
+    vlm_load_4bit=True,
 )
 # Launch the app
 app.launch()