From 7c7d0326c5c0e5135e7bfe72d67872f10a325a4a Mon Sep 17 00:00:00 2001
From: Xinyu Dong <dongxinyu23@gmail.com>
Date: Mon, 15 Dec 2025 21:21:28 +0800
Subject: [PATCH] [Model] registry llama.py to vLLM

---
 vllm_kunlun/models/__init__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/vllm_kunlun/models/__init__.py b/vllm_kunlun/models/__init__.py
index 4eb7de1..7dd089c 100644
--- a/vllm_kunlun/models/__init__.py
+++ b/vllm_kunlun/models/__init__.py
@@ -78,6 +78,10 @@ def register_model():
         "SeedOssForCausalLM",
         "vllm_kunlun.models.seed_oss:SeedOssForCausalLM")
 
+    ModelRegistry.register_model(
+        "LlamaForCausalLM",
+        "vllm_kunlun.models.llama:LlamaForCausalLM")
+
 
 def register_quant_method():
-    """to do"""
\ No newline at end of file
+    """to do"""