Skip to content

Commit

Permalink
New PR for Faster Whisper: Batching Support, Speed Boosts, and Qualit…
Browse files Browse the repository at this point in the history
…y Enhancements (SYSTRAN#856)

Batching Support, Speed Boosts, and Quality Enhancements

---------

Co-authored-by: Hargun Mujral <83234565+hargunmujral@users.noreply.github.com>
Co-authored-by: MahmoudAshraf97 <hassouna97.ma@gmail.com>
  • Loading branch information
3 people committed Oct 2, 2024
1 parent 1c22ff3 commit 5b2cc66
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions tests/test_transcribe.py
Original file line number Diff line number Diff line change
Expand Up @@ -145,6 +145,13 @@ def test_stereo_diarization(data_dir):
assert transcription == "The horizon seems extremely distant."


def test_multisegment_lang_id(physcisworks_path):
model = WhisperModel("tiny")
language_info = model.detect_language_multi_segment(physcisworks_path)
assert language_info["language_code"] == "en"
assert language_info["language_confidence"] > 0.8


def test_suppressed_tokens_minus_1():
model = WhisperModel("tiny.en")

Expand Down

0 comments on commit 5b2cc66

Please sign in to comment.