Add Baichuan-7B to README (#494)

This commit is contained in:
Zhuohan Li 2023-07-25 15:25:12 -07:00 committed by GitHub
parent 2d867b55fa
commit df5dd3c68e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 0 deletions

View File

@ -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.)

View File

@ -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.