From f177900fb02eb19b00679307f09f22a8bf9dd75c Mon Sep 17 00:00:00 2001 From: jlqian98 Date: Thu, 19 Mar 2026 13:29:54 +0800 Subject: [PATCH] fix: avoid shape mismatch when merging segments in svs inference --- cli/inference.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cli/inference.py b/cli/inference.py index 148cf73..95c219d 100644 --- a/cli/inference.py +++ b/cli/inference.py @@ -114,7 +114,8 @@ def process(args, config, model: torch.nn.Module): ) generated_audio = generated_audio.squeeze().cpu().numpy() - generated_merged[start_sample_idx : start_sample_idx + generated_audio.shape[0]] = generated_audio + gen_len = min(generated_audio.shape[0], generated_merged.shape[0] - start_sample_idx) + generated_merged[start_sample_idx: start_sample_idx + gen_len] = generated_audio[:gen_len] merged_path = os.path.join(args.save_dir, "generated.wav") sf.write(merged_path, generated_merged, 24000)