Skip to content

[Cohere] Enable Cohere Transcribe - #38120

Merged
ywang96 merged 6 commits into
vllm-project:mainfrom
ekagra-ranjan:er-asr-cohere-release-on
Mar 25, 2026
Merged

ywang96 merged 6 commits into
vllm-project:mainfrom
ekagra-ranjan:er-asr-cohere-release-on

Conversation

@ekagra-ranjan

@ekagra-ranjan ekagra-ranjan commented Mar 25, 2026 •

Copy link
Copy Markdown
Contributor

Followup to #35809
Turns on Cohere Transcribe model (official name) refered as CohereAsrForConditionalGeneration withing vLLM code. The official HF repo will be CohereLabs/cohere-transcribe-03-2026.

The PR also adds the model to test_transcription_api_correctness.py but doesnt enable it yet. Its adds additional value to this test because this model touches all the recent changes that was done to the variable length encoder inputs. Earlier, vLLM only had padded length encoder inputs which was suited for whisper but not for Cohere-Transcribe.

That test calls a normalizer from the tokenizer but not all the models have a normalizer so we use the standard normalizer in the test for all the models in that test.

Cmd to run the model:
Install vLLM nightly and dependencies

uv pip install -U vllm --torch-backend=auto --extra-index-url https://wheels.vllm.ai/nightly
uv pip install vllm[audio]
uv pip install librosa

Start server

vllm serve CohereLabs/cohere-transcribe-03-2026 --trust-remote-code

Send request

AUDIO_PATH=<your-audio-path>

curl -v -X POST http://localhost:8000/v1/audio/transcriptions \
-H "Authorization: Bearer $VLLM_API_KEY" \
-F "file=@$(realpath ${AUDIO_PATH})" \
-F "model=CohereLabs/cohere-transcribe-03-2026"

Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
@mergify

mergify Bot commented Mar 25, 2026

Copy link
Copy Markdown
Contributor

Documentation preview: https://vllm--38120.org.readthedocs.build/en/38120/

@mergify mergify Bot added documentation Improvements or additions to documentation new-model Requests to new models labels Mar 25, 2026

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request integrates the Cohere-Transcribe model by updating model paths, enabling its online availability, adding it to the supported models documentation, and incorporating it into transcription API correctness tests. The tests now utilize a standardized English text normalizer from Whisper for evaluation. A review comment suggests refactoring the model registry entry for CohereASRForConditionalGeneration to avoid duplication and improve maintainability.

Comment thread vllm/model_executor/models/registry.py
@mergify

mergify Bot commented Mar 25, 2026

Copy link
Copy Markdown
Contributor

Hi @ekagra-ranjan, the pre-commit checks have failed. Please run:

uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-files

Then, commit the changes and push to your branch.

For future commits, pre-commit will run automatically on changed files before each commit.

Tip

Is mypy failing?
mypy is run differently in CI. If the failure is related to this check, please use the following command to run it locally:
# For mypy (substitute "3.10" with the failing version if needed)
pre-commit run --hook-stage manual mypy-3.10

@ywang96 ywang96 added the ready ONLY add when PR is ready to merge/full CI is needed label Mar 25, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
@ekagra-ranjan

Copy link
Copy Markdown
Contributor Author

Some CI is failing due to the model name issue. I'll fix it

…neration

Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
@ywang96
ywang96 merged commit 7b54f60 into vllm-project:main Mar 25, 2026
57 of 59 checks passed
@ekagra-ranjan ekagra-ranjan changed the title [Cohere] Enable Cohere-Transcribe [Cohere] Enable Cohere Transcribe Mar 26, 2026
RhizoNymph pushed a commit to RhizoNymph/vllm that referenced this pull request Mar 26, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
nithinvc pushed a commit to nithinvc/vllm that referenced this pull request Mar 27, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>

Signed-off-by: Nithin Chalapathi <nithin.ch10@gmail.com>
JiantaoXu pushed a commit to JiantaoXu/vllm that referenced this pull request Mar 28, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
puririshi98 pushed a commit to puririshi98/vllm that referenced this pull request Apr 7, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
Signed-off-by: Rishi Puri <riship@nvidia.com>
mtparet pushed a commit to blackfuel-ai/vllm that referenced this pull request Apr 9, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
mystous pushed a commit to mystous/vllm_hybrid that referenced this pull request May 10, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
my-other-github-account pushed a commit to my-other-github-account/vllm that referenced this pull request May 15, 2026
Signed-off-by: Ekagra Ranjan <3116519+ekagra-ranjan@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or additions to documentation new-model Requests to new models ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants