Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update whisper.cc : Add logits to WhisperGenerationResult #1846

Merged
merged 2 commits into from
Feb 10, 2025

Conversation

sammygrey
Copy link
Contributor

@sammygrey sammygrey commented Jan 16, 2025

WhisperGenerationResult was missing return value for logits despite register_whisper having a readonly definition for said logits.
Logits are documented on https://opennmt.net/CTranslate2/python/ctranslate2.models.Whisper.html as being returned when return_logits_vocab is set to True, however, there was no written return value to access them within WhisperGenerationResult if it was set to True.

This fixes issue #1779.

Added logits to WhisperGenerationResult
@sammygrey sammygrey changed the title Update whisper.cc : Added logits to WhisperGenerationResult Update whisper.cc : Add logits to WhisperGenerationResult Feb 1, 2025
@MahmoudAshraf97
Copy link
Contributor

also fixes #1822

@minhthuc2502
Copy link
Collaborator

Thank you for your fix. I will merge this

@minhthuc2502 minhthuc2502 merged commit 6a3dc63 into OpenNMT:master Feb 10, 2025
13 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants