fix

llava/llava_agent.py

@@ -22,10 +22,16 @@ class LLavaAgent:
             device_map = {'model': torch.device(self.device).index, 'lm_head': torch.device(self.device).index}
         else:
             device_map = 'auto'
-        model_path = os.path.expanduser(model_path)
-        model_name = get_model_name_from_path(model_path)
+
+        # Directly use HF repo if not local
+        if os.path.exists(model_path):
+            resolved_path = model_path
+        else:
+            resolved_path = model_path  # treat as HF model ID
+
+        model_name = get_model_name_from_path(resolved_path)
         tokenizer, model, image_processor, context_len = load_pretrained_model(
-            model_path, None, model_name, device=self.device, device_map=device_map,
+            resolved_path, None, model_name, device=self.device, device_map=device_map,
             load_8bit=load_8bit, load_4bit=load_4bit)
         self.model = model
         self.image_processor = image_processor