From 471a7a456667971721a1ce589029fc82be6ab193 Mon Sep 17 00:00:00 2001
From: BasicCoder <abasiccoder@gmail.com>
Date: Tue, 27 Jun 2023 00:23:57 +0800
Subject: [PATCH] Compatible with Decapoda Research llama hf version (#251)

---
 vllm/model_executor/model_loader.py | 1 +
 1 file changed, 1 insertion(+)

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,
 }