From 0999a613c56c462b063b6b25d96260e1fc6ee2de Mon Sep 17 00:00:00 2001 From: atila Date: Thu, 4 Apr 2024 08:45:52 +0900 Subject: [PATCH] Allow token timestamps for distil-large-v3 --- scripts/generate_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/generate_model.py b/scripts/generate_model.py index 7987d98..8ff08ad 100644 --- a/scripts/generate_model.py +++ b/scripts/generate_model.py @@ -98,7 +98,7 @@ def cli(): # FIXME(atiorh): Remove this once distil-whisper-* models are updated args.disable_token_timestamps = False - if "distil-whisper" in args.model_version: + if "distil" in args.model_version and "distil-large-v3" not in args.model_version: logger.info( "Disabling token-level timestamps due to missing alignment_heads in distil-whisper-* models" )