Skip to content

Commit 3b1e4c6

Browse files
authored
[Docs] Add GPT2ForSequenceClassification to supported models in docs (#19932)
Signed-off-by: nie3e <adrcwiek@gmail.com>
1 parent 2c5302f commit 3b1e4c6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/models/supported_models.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -445,7 +445,7 @@ Specified using `--task classify`.
445445
| Architecture | Models | Example HF Models | [LoRA][lora-adapter] | [PP][distributed-serving] | [V1](gh-issue:8779) |
446446
|----------------------------------|----------|----------------------------------------|------------------------|-----------------------------|-----------------------|
447447
| `JambaForSequenceClassification` | Jamba | `ai21labs/Jamba-tiny-reward-dev`, etc. | ✅︎ | ✅︎ | |
448-
448+
| `GPT2ForSequenceClassification` | GPT2 | `nie3e/sentiment-polish-gpt2-small` | | | |
449449
If your model is not in the above list, we will try to automatically convert the model using
450450
[as_classification_model][vllm.model_executor.models.adapters.as_classification_model]. By default, the class probabilities are extracted from the softmaxed hidden state corresponding to the last token.
451451

0 commit comments

Comments
 (0)