From df5dd3c68ec7007e4b3a27907f4d954f1c7d511d Mon Sep 17 00:00:00 2001
From: Zhuohan Li <zhuohan123@gmail.com>
Date: Tue, 25 Jul 2023 15:25:12 -0700
Subject: [PATCH] Add Baichuan-7B to README (#494)

---
 README.md                               | 1 +
 docs/source/models/supported_models.rst | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/README.md b/README.md
index d26e7acb4f88d..f67e6b6c431e0 100644
--- a/README.md
+++ b/README.md
@@ -42,6 +42,7 @@ vLLM is flexible and easy to use with:
 
 vLLM seamlessly supports many Huggingface models, including the following architectures:
 
+- Baichuan-7B (`baichuan-inc/Baichuan-7B`)
 - BLOOM (`bigscience/bloom`, `bigscience/bloomz`, etc.)
 - GPT-2 (`gpt2`, `gpt2-xl`, etc.)
 - GPT BigCode (`bigcode/starcoder`, `bigcode/gpt_bigcode-santacoder`, etc.)
diff --git a/docs/source/models/supported_models.rst b/docs/source/models/supported_models.rst
index ea9cbcab7fae7..4c9a47e7d0b1e 100644
--- a/docs/source/models/supported_models.rst
+++ b/docs/source/models/supported_models.rst
@@ -14,6 +14,9 @@ Alongside each architecture, we include some popular models that use it.
   * - Architecture
     - Models
     - Example HuggingFace Models
+  * - :code:`BaiChuanForCausalLM`
+    - Baichuan-7B
+    - :code:`baichuan-inc/Baichuan-7B`.
   * - :code:`BloomForCausalLM`
     - BLOOM, BLOOMZ, BLOOMChat
     - :code:`bigscience/bloom`, :code:`bigscience/bloomz`, etc.