Skip to content

Commit a9c8212

Browse files
authored
[FIX] Add Gemma model to the doc (#2966)
1 parent c20ecb6 commit a9c8212

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ vLLM seamlessly supports many Hugging Face models, including the following archi
6060
- ChatGLM (`THUDM/chatglm2-6b`, `THUDM/chatglm3-6b`, etc.)
6161
- DeciLM (`Deci/DeciLM-7B`, `Deci/DeciLM-7B-instruct`, etc.)
6262
- Falcon (`tiiuae/falcon-7b`, `tiiuae/falcon-40b`, `tiiuae/falcon-rw-7b`, etc.)
63+
- Gemma (`google/gemma-2b`, `google/gemma-7b`, etc.)
6364
- GPT-2 (`gpt2`, `gpt2-xl`, etc.)
6465
- GPT BigCode (`bigcode/starcoder`, `bigcode/gpt_bigcode-santacoder`, etc.)
6566
- GPT-J (`EleutherAI/gpt-j-6b`, `nomic-ai/gpt4all-j`, etc.)

docs/source/models/supported_models.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,9 @@ Alongside each architecture, we include some popular models that use it.
3232
* - :code:`FalconForCausalLM`
3333
- Falcon
3434
- :code:`tiiuae/falcon-7b`, :code:`tiiuae/falcon-40b`, :code:`tiiuae/falcon-rw-7b`, etc.
35+
* - :code:`GemmaForCausalLM`
36+
- Gemma
37+
- :code:`google/gemma-2b`, :code:`google/gemma-7b`, etc.
3538
* - :code:`GPT2LMHeadModel`
3639
- GPT-2
3740
- :code:`gpt2`, :code:`gpt2-xl`, etc.

0 commit comments

Comments
 (0)