Skip to content

GitLab

Unverified Commit ebe56a00 authored by dongluw's avatar dongluw Committed by GitHub
Browse files

Small fix for Command-A-Vision (#23268) 


Signed-off-by: default avatardonglu <donglu@cohere.com>
parent f77a0802
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/model_executor/models/cohere2_vision.py
View file @ ebe56a00
...@@ -348,7 +348,7 @@ class Cohere2VisionForConditionalGeneration(nn.Module, SupportsMultiModal, ...@@ -348,7 +348,7 @@ class Cohere2VisionForConditionalGeneration(nn.Module, SupportsMultiModal,
vllm_config=vllm_config, vllm_config=vllm_config,
hf_config=config.text_config, hf_config=config.text_config,
prefix=maybe_prefix(prefix, "language_model"), prefix=maybe_prefix(prefix, "language_model"),
architectures=["Cohere2ForCausalLM"]) architectures=config.text_config.architectures)
@property @property
def dtype(self): def dtype(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment