feat(pre-sd): add leading zeros for 4-digit width of the output file …

…name's numeric part #1154 (#1155)
voicepaw · Apr 11, 2024 · 41b147f · 41b147f
1 parent 50f6d79
commit 41b147f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/so_vits_svc_fork/preprocessing/preprocess_speaker_diarization.py b/src/so_vits_svc_fork/preprocessing/preprocess_speaker_diarization.py
@@ -52,7 +52,7 @@ def _process_one(
         speaker_count[speaker] += 1
         audio_cut = audio[int(segment.start * sr) : int(segment.end * sr)]
         sf.write(
-            (output_dir / f"{speaker}_{speaker_count[speaker]}.wav"),
+            (output_dir / f"{speaker}_{speaker_count[speaker]:04d}.wav"),
             audio_cut,
             sr,
         )