Hugging Face's logo

Spaces:
Yiyuan
/
VSA
Runtime error

App Files Community
1
cnzzx commited on
Commit
781fb6b
·
1 Parent(s): 7b85afa

update

Browse files
Files changed (1) hide show
  1. models/vsa_model.py +1 -1
models/vsa_model.py CHANGED
@@ -306,7 +306,7 @@ class VisionSearchAssistant:
306
  """
307
  def __init__(
308
  self,
309
- search_model: str = "internlm/internlm2_5-7b-chat",
310
  ground_model: str = "IDEA-Research/grounding-dino-base",
311
  ground_device: str = "cuda:1",
312
  vlm_model: str = "liuhaotian/llava-v1.6-vicuna-7b",
 
306
  """
307
  def __init__(
308
  self,
309
+ search_model: str = "internlm/internlm2_5-1_8b-chat",
310
  ground_model: str = "IDEA-Research/grounding-dino-base",
311
  ground_device: str = "cuda:1",
312
  vlm_model: str = "liuhaotian/llava-v1.6-vicuna-7b",