From 221f3cbad82afe32d8c9424bda6d82319b475f0c Mon Sep 17 00:00:00 2001 From: laurent Date: Tue, 8 Jul 2025 12:18:28 +0200 Subject: [PATCH] Avoid the config override for the extra-heads. --- scripts/stt_from_file_pytorch.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/scripts/stt_from_file_pytorch.py b/scripts/stt_from_file_pytorch.py index 46479ea..4af89ff 100644 --- a/scripts/stt_from_file_pytorch.py +++ b/scripts/stt_from_file_pytorch.py @@ -138,13 +138,9 @@ def main(args): mimi = info.get_mimi(device=args.device) tokenizer = info.get_text_tokenizer() - lm_kwargs_overrides = {} - if args.vad: - lm_kwargs_overrides = {"extra_heads_num_heads": 4} lm = info.get_moshi( device=args.device, dtype=torch.bfloat16, - lm_kwargs_overrides=lm_kwargs_overrides, ) lm_gen = moshi.models.LMGen(lm, temp=0, temp_text=0.0)