From 65bb0e7733cb319d7aeedec3046680dc6ec4aa8f Mon Sep 17 00:00:00 2001 From: DarkLight1337 Date: Fri, 4 Jul 2025 12:57:02 +0000 Subject: [PATCH] [Doc] Fix classification table in list of supported models Signed-off-by: DarkLight1337 --- docs/models/supported_models.md | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/models/supported_models.md b/docs/models/supported_models.md index cf062311b00c..23d71fd44525 100644 --- a/docs/models/supported_models.md +++ b/docs/models/supported_models.md @@ -470,6 +470,7 @@ Specified using `--task classify`. |----------------------------------|----------|----------------------------------------|------------------------|-----------------------------|-----------------------| | `JambaForSequenceClassification` | Jamba | `ai21labs/Jamba-tiny-reward-dev`, etc. | ✅︎ | ✅︎ | | | `GPT2ForSequenceClassification` | GPT2 | `nie3e/sentiment-polish-gpt2-small` | | | ✅︎ | + If your model is not in the above list, we will try to automatically convert the model using [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.