update

models/vsa_model.py

@@ -126,7 +126,7 @@ class VLM:
         load_8bit: bool = False,
         load_4bit: bool = True,
         temperature: float = 0.2,
-        max_new_tokens: int = 1024,
+        max_new_tokens: int = 2000,
     ):
         disable_torch_init()
         model_name = get_model_name_from_path(model_path)
@@ -307,7 +307,7 @@ class VisionSearchAssistant:
     def __init__(
         self,
         search_model: str = "internlm/internlm2_5-1_8b-chat",
-        ground_model: str = "IDEA-Research/grounding-dino-base",
+        ground_model: str = "IDEA-Research/grounding-dino-tiny",
         ground_device: str = "cuda:1",
         vlm_model: str = "liuhaotian/llava-v1.6-vicuna-7b",
         vlm_device: str = "cuda:2",
@@ -324,7 +324,8 @@ class VisionSearchAssistant:
         self.use_correlate = True
 
         self.searcher = WebSearcher(
-            model_path = self.search_model
+            model_path = self.search_model,
+            lang = 'en'
         )
         self.grounder = VisualGrounder(
             model_path = self.ground_model,