Compatible with Decapoda Research llama hf version (#251)

2025-12-10 06:25:01 +08:00 · 2023-06-27 00:23:57 +08:00 · 2023-06-27 00:23:57 +08:00 · 471a7a4566
commit 471a7a4566
parent 6214dd6ce9
1 changed files with 1 additions and 0 deletions
--- 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,
 }