[Doc] Add Baichuan 13B to supported models (#656)

Author: zhuohan123

README.md

@@ -42,7 +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`)
+- Baichuan (`baichuan-inc/Baichuan-7B`, `baichuan-inc/Baichuan-13B-Chat`, etc.)
 - BLOOM (`bigscience/bloom`, `bigscience/bloomz`, etc.)
 - Falcon (`tiiuae/falcon-7b`, `tiiuae/falcon-40b`, `tiiuae/falcon-rw-7b`, etc.)
 - GPT-2 (`gpt2`, `gpt2-xl`, etc.)

docs/source/models/supported_models.rst

@@ -15,8 +15,8 @@ Alongside each architecture, we include some popular models that use it.
     - Models
     - Example HuggingFace Models
   * - :code:`BaiChuanForCausalLM`
-    - Baichuan-7B
-    - :code:`baichuan-inc/Baichuan-7B`.
+    - Baichuan
+    - :code:`baichuan-inc/Baichuan-7B`, `baichuan-inc/Baichuan-13B-Chat`, etc.
   * - :code:`BloomForCausalLM`
     - BLOOM, BLOOMZ, BLOOMChat
     - :code:`bigscience/bloom`, :code:`bigscience/bloomz`, etc.

vllm/model_executor/models/__init__.py

@@ -1,4 +1,5 @@
-from vllm.model_executor.models.baichuan import BaiChuanForCausalLM, BaichuanForCausalLM
+from vllm.model_executor.models.baichuan import (BaiChuanForCausalLM,
+                                                 BaichuanForCausalLM)
 from vllm.model_executor.models.bloom import BloomForCausalLM
 from vllm.model_executor.models.falcon import FalconForCausalLM
 from vllm.model_executor.models.gpt2 import GPT2LMHeadModel