Skip to content

Commit bc092ea

Browse files
jacobkahnywang96
andauthored
Map CwmForCausalLM to llama and LlamaForCausalLM (vllm-project#25611)
Signed-off-by: Jacob Kahn <jacobkahn1@gmail.com> Co-authored-by: Roger Wang <hey@rogerw.io>
1 parent 755ed7b commit bc092ea

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tests/models/registry.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -196,6 +196,9 @@ def check_available_online(
196196
trust_remote_code=True),
197197
"Cohere2ForCausalLM": _HfExamplesInfo("CohereForAI/c4ai-command-r7b-12-2024", # noqa: E501
198198
trust_remote_code=True),
199+
"CwmForCausalLM": _HfExamplesInfo("facebook/cwm", # noqa: E501
200+
trust_remote_code=True,
201+
is_available_online=False),
199202
"DbrxForCausalLM": _HfExamplesInfo("databricks/dbrx-instruct"),
200203
"DeciLMForCausalLM": _HfExamplesInfo("nvidia/Llama-3_3-Nemotron-Super-49B-v1", # noqa: E501
201204
trust_remote_code=True),

vllm/model_executor/models/registry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@
6464
"ChatGLMForConditionalGeneration": ("chatglm", "ChatGLMForCausalLM"),
6565
"CohereForCausalLM": ("commandr", "CohereForCausalLM"),
6666
"Cohere2ForCausalLM": ("commandr", "CohereForCausalLM"),
67+
"CwmForCausalLM": ("llama", "LlamaForCausalLM"),
6768
"DbrxForCausalLM": ("dbrx", "DbrxForCausalLM"),
6869
"DeciLMForCausalLM": ("nemotron_nas", "DeciLMForCausalLM"),
6970
"DeepseekForCausalLM": ("deepseek", "DeepseekForCausalLM"),

0 commit comments

Comments
 (0)