remove hardcoded local model

dnth · Oct 23, 2024 · 289a353 · 289a353
1 parent 78c364b
commit 289a353
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/xinfer/vllm/molmo.py b/xinfer/vllm/molmo.py
@@ -20,8 +20,7 @@ def __init__(
 
     def load_model(self, **kwargs):
         self.model = LLM(
-            # model=self.model_id,
-            model="/home/dnth/Desktop/cv-docker-images/image_captioning/molmo/molmo_7b_d_0924",
+            model=self.model_id,
             trust_remote_code=True,
             dtype=self.dtype,
             **kwargs,