Compatible with Decapoda Research llama hf version (#251)

471a7a45 · BasicCoder · GitHub · 6214dd6c · 471a7a45
Unverified Commit 471a7a45 authored Jun 27, 2023 by BasicCoder Committed by GitHub Jun 26, 2023
Show 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
@@ -16,6 +16,7 @@ _MODEL_REGISTRY = {
    "GPTBigCodeForCausalLM": GPTBigCodeForCausalLM,
    "GPTNeoXForCausalLM": GPTNeoXForCausalLM,
    "LlamaForCausalLM": LlamaForCausalLM,
+    "LLaMAForCausalLM": LlamaForCausalLM,
    "OPTForCausalLM": OPTForCausalLM,
 }