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
......@@ -348,7 +348,7 @@ class Cohere2VisionForConditionalGeneration(nn.Module, SupportsMultiModal,
vllm_config=vllm_config,
hf_config=config.text_config,
prefix=maybe_prefix(prefix, "language_model"),
architectures=["Cohere2ForCausalLM"])
architectures=config.text_config.architectures)
@property
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