Hotfix: pass through model revision in `VlmCausalLM` (#2258)
This commit is contained in:
parent
3b41e93a09
commit
3f37a66774
|
@ -261,7 +261,12 @@ class VlmCausalLM(FlashCausalLM):
|
|||
**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]:
|
||||
|
|
Loading…
Reference in New Issue