Allow token timestamps for distil-large-v3

argmaxinc · Apr 3, 2024 · 0999a61 · 0999a61
1 parent 5cfec57
commit 0999a61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"
         )