Hotfix: pass through model revision in VlmCausalLM (#2258)

huggingface · Jul 26, 2024 · 0f59eb8 · 0f59eb8
1 parent ede341d
commit 0f59eb8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/server/text_generation_server/models/vlm_causal_lm.py b/server/text_generation_server/models/vlm_causal_lm.py
@@ -261,7 +261,12 @@ def __init__(
             **processor_kwargs,
         )
         self.batch_class = batch_class
-        super().__init__(model_id=model_id, **kwargs)
+        super().__init__(
+            model_id=model_id,
+            revision=revision,
+            trust_remote_code=trust_remote_code,
+            **kwargs,
+        )
 
     @property
     def batch_type(self) -> Type[VlmCausalLMBatch]: