Support ChatGLMForConditionalGeneration (vllm-project#1932)

Co-authored-by: shujunhua1 <[email protected]>
EmbeddedLLM · Dec 5, 2023 · 1d9b737 · 1d9b737
1 parent 60dc62d
commit 1d9b737
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vllm/model_executor/model_loader.py b/vllm/model_executor/model_loader.py
@@ -19,6 +19,7 @@
     "BaichuanForCausalLM": BaichuanForCausalLM,  # baichuan-13b
     "BloomForCausalLM": BloomForCausalLM,
     "ChatGLMModel": ChatGLMForCausalLM,
+    "ChatGLMForConditionalGeneration": ChatGLMForCausalLM,
     "FalconForCausalLM": FalconForCausalLM,
     "GPT2LMHeadModel": GPT2LMHeadModel,
     "GPTBigCodeForCausalLM": GPTBigCodeForCausalLM,