Support ChatGLMForConditionalGeneration (#1932)

Co-authored-by: shujunhua1 <shujunhua1@jd.com>

Support ChatGLMForConditionalGeneration (#1932)
Co-authored-by: shujunhua1 <shujunhua1@jd.com>
1d9b737e · dancingpipi · GitHub · 60dc62dc · 1d9b737e
Unverified Commit 1d9b737e authored Dec 06, 2023 by dancingpipi Committed by GitHub Dec 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

vllm/model_executor/model_loader.py vllm/model_executor/model_loader.py +1 -0

No files found.
--- a/vllm/model_executor/model_loader.py
+++ b/vllm/model_executor/model_loader.py
@@ -19,6 +19,7 @@ _MODEL_REGISTRY = {
    "BaichuanForCausalLM": BaichuanForCausalLM,  # baichuan-13b
    "BloomForCausalLM": BloomForCausalLM,
    "ChatGLMModel": ChatGLMForCausalLM,
+    "ChatGLMForConditionalGeneration": ChatGLMForCausalLM,
    "FalconForCausalLM": FalconForCausalLM,
    "GPT2LMHeadModel": GPT2LMHeadModel,
    "GPTBigCodeForCausalLM": GPTBigCodeForCausalLM,