Skip to content

Commit 901a6c2

Browse files
DarkLight1337huydhn
authored andcommitted
[Doc] Fix classification table in list of supported models (vllm-project#20489)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
1 parent 2a6dd9d commit 901a6c2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

docs/models/supported_models.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -470,6 +470,7 @@ Specified using `--task classify`.
470470
|----------------------------------|----------|----------------------------------------|------------------------|-----------------------------|-----------------------|
471471
| `JambaForSequenceClassification` | Jamba | `ai21labs/Jamba-tiny-reward-dev`, etc. | ✅︎ | ✅︎ | |
472472
| `GPT2ForSequenceClassification` | GPT2 | `nie3e/sentiment-polish-gpt2-small` | | | ✅︎ |
473+
473474
If your model is not in the above list, we will try to automatically convert the model using
474475
[as_seq_cls_model][vllm.model_executor.models.adapters.as_seq_cls_model]. By default, the class probabilities are extracted from the softmaxed hidden state corresponding to the last token.
475476

0 commit comments

Comments
 (0)