diff --git a/xinfer/vllm/molmo.py b/xinfer/vllm/molmo.py index 4aa1d90..a89b85e 100644 --- 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,