Compatible with Decapoda Research llama hf version (#251)

2023-06-27 00:23:57 +08:00
parent 6214dd6ce9
commit 471a7a4566
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,
 }