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

Signed-off-by: donglu <donglu@cohere.com>
2025-08-20 14:15:18 -04:00
parent f77a0802b7
commit ebe56a0064
1 changed files with 1 additions and 1 deletions
--- a/vllm/model_executor/models/cohere2_vision.py
+++ b/vllm/model_executor/models/cohere2_vision.py
@@ -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):