Skip to content

Commit 88cd5d1

Browse files
committed
refactor: s/kwargs.pop("task")/kwargs.pop("task", None)
1 parent f4a44b9 commit 88cd5d1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

daras_ai_v2/asr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1297,9 +1297,9 @@ def run_asr(
12971297
kwargs["chunk_length_s"] = 60
12981298
kwargs["stride_length_s"] = (6, 0)
12991299
kwargs["batch_size"] = 32
1300-
elif "akera" in selected_model.name:
1300+
elif selected_model == AsrModels.whisper_akera_large_v3:
13011301
# don't pass language or task
1302-
kwargs.pop("task")
1302+
kwargs.pop("task", None)
13031303
kwargs["max_length"] = 448
13041304
elif "whisper" in selected_model.name:
13051305
forced_lang = forced_asr_languages.get(selected_model)

0 commit comments

Comments
 (0)