Skip to content

GitLab

Unverified Commit 1d9b737e authored by dancingpipi's avatar dancingpipi Committed by GitHub
Browse files

Support ChatGLMForConditionalGeneration (#1932) 


Co-authored-by: default avatarshujunhua1 <shujunhua1@jd.com>
parent 60dc62dc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
vllm/model_executor/model_loader.py
View file @ 1d9b737e
...@@ -19,6 +19,7 @@ _MODEL_REGISTRY = { ...@@ -19,6 +19,7 @@ _MODEL_REGISTRY = {
"BaichuanForCausalLM": BaichuanForCausalLM, # baichuan-13b "BaichuanForCausalLM": BaichuanForCausalLM, # baichuan-13b
"BloomForCausalLM": BloomForCausalLM, "BloomForCausalLM": BloomForCausalLM,
"ChatGLMModel": ChatGLMForCausalLM, "ChatGLMModel": ChatGLMForCausalLM,
"ChatGLMForConditionalGeneration": ChatGLMForCausalLM,
"FalconForCausalLM": FalconForCausalLM, "FalconForCausalLM": FalconForCausalLM,
"GPT2LMHeadModel": GPT2LMHeadModel, "GPT2LMHeadModel": GPT2LMHeadModel,
"GPTBigCodeForCausalLM": GPTBigCodeForCausalLM, "GPTBigCodeForCausalLM": GPTBigCodeForCausalLM,
......
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