Skip to content

Commit c9eef37

Browse files
authored
[Model] Initial Support for Chameleon (#5770)
1 parent 396d92d commit c9eef37

File tree

5 files changed

+665
-4
lines changed

5 files changed

+665
-4
lines changed

vllm/model_executor/models/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@
1616
"BaiChuanForCausalLM": ("baichuan", "BaiChuanForCausalLM"), # baichuan-7b
1717
"BaichuanForCausalLM": ("baichuan", "BaichuanForCausalLM"), # baichuan-13b
1818
"BloomForCausalLM": ("bloom", "BloomForCausalLM"),
19+
"ChameleonForCausalLM":
20+
("chameleon", "ChameleonForConditionalGeneration"
21+
), #TODO(ywang96): fix model name when huggingface fixes it
1922
"ChatGLMModel": ("chatglm", "ChatGLMForCausalLM"),
2023
"ChatGLMForConditionalGeneration": ("chatglm", "ChatGLMForCausalLM"),
2124
"CohereForCausalLM": ("commandr", "CohereForCausalLM"),

0 commit comments

Comments
 (0)