diff --git a/vllm/model_executor/model_loader.py b/vllm/model_executor/model_loader.py index 93faa116c68a9..7424104fa2c9b 100644 --- a/vllm/model_executor/model_loader.py +++ b/vllm/model_executor/model_loader.py @@ -16,6 +16,7 @@ _MODEL_REGISTRY = { "GPTBigCodeForCausalLM": GPTBigCodeForCausalLM, "GPTNeoXForCausalLM": GPTNeoXForCausalLM, "LlamaForCausalLM": LlamaForCausalLM, + "LLaMAForCausalLM": LlamaForCausalLM, "OPTForCausalLM": OPTForCausalLM, }