Skip to content

GitLab

Unverified Commit bb706d60 authored by Yejing Lai's avatar Yejing Lai Committed by GitHub
Browse files

Fix TeleChatForCausalLM not register issue (#29473) 


Signed-off-by: default avatarLai, Yejing <yejing.lai@intel.com>
parent e30859df
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
tests/models/registry.py
View file @ bb706d60
...@@ -436,6 +436,9 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { ...@@ -436,6 +436,9 @@ _TEXT_GENERATION_EXAMPLE_MODELS = {
"SolarForCausalLM": _HfExamplesInfo( "SolarForCausalLM": _HfExamplesInfo(
"upstage/solar-pro-preview-instruct", trust_remote_code=True "upstage/solar-pro-preview-instruct", trust_remote_code=True
), ),
"TeleChatForCausalLM": _HfExamplesInfo(
"chuhac/TeleChat2-35B", trust_remote_code=True
),
"TeleChat2ForCausalLM": _HfExamplesInfo( "TeleChat2ForCausalLM": _HfExamplesInfo(
"Tele-AI/TeleChat2-3B", trust_remote_code=True "Tele-AI/TeleChat2-3B", trust_remote_code=True
), ),
......
vllm/model_executor/models/registry.py
View file @ bb706d60
...@@ -170,6 +170,7 @@ _TEXT_GENERATION_MODELS = { ...@@ -170,6 +170,7 @@ _TEXT_GENERATION_MODELS = {
"StableLmForCausalLM": ("stablelm", "StablelmForCausalLM"), "StableLmForCausalLM": ("stablelm", "StablelmForCausalLM"),
"Starcoder2ForCausalLM": ("starcoder2", "Starcoder2ForCausalLM"), "Starcoder2ForCausalLM": ("starcoder2", "Starcoder2ForCausalLM"),
"SolarForCausalLM": ("solar", "SolarForCausalLM"), "SolarForCausalLM": ("solar", "SolarForCausalLM"),
"TeleChatForCausalLM": ("telechat2", "TeleChat2ForCausalLM"),
"TeleChat2ForCausalLM": ("telechat2", "TeleChat2ForCausalLM"), "TeleChat2ForCausalLM": ("telechat2", "TeleChat2ForCausalLM"),
"TeleFLMForCausalLM": ("teleflm", "TeleFLMForCausalLM"), "TeleFLMForCausalLM": ("teleflm", "TeleFLMForCausalLM"),
"XverseForCausalLM": ("llama", "LlamaForCausalLM"), "XverseForCausalLM": ("llama", "LlamaForCausalLM"),
...@@ -207,6 +208,7 @@ _EMBEDDING_MODELS = { ...@@ -207,6 +208,7 @@ _EMBEDDING_MODELS = {
"Qwen2ForProcessRewardModel": ("qwen2_rm", "Qwen2ForProcessRewardModel"), "Qwen2ForProcessRewardModel": ("qwen2_rm", "Qwen2ForProcessRewardModel"),
"RobertaForMaskedLM": ("roberta", "RobertaEmbeddingModel"), "RobertaForMaskedLM": ("roberta", "RobertaEmbeddingModel"),
"RobertaModel": ("roberta", "RobertaEmbeddingModel"), "RobertaModel": ("roberta", "RobertaEmbeddingModel"),
"TeleChatForCausalLM": ("telechat2", "TeleChat2ForCausalLM"),
"TeleChat2ForCausalLM": ("telechat2", "TeleChat2ForCausalLM"), "TeleChat2ForCausalLM": ("telechat2", "TeleChat2ForCausalLM"),
"XLMRobertaModel": ("roberta", "RobertaEmbeddingModel"), "XLMRobertaModel": ("roberta", "RobertaEmbeddingModel"),
# [Multimodal] # [Multimodal]
......
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