Update infer.py

2025-09-03 22:24:19 +00:00 · 2023-12-04 10:01:38 +08:00 · 2023-12-04 10:01:38 +08:00 · 0cb2b12adb
commit 0cb2b12adb
parent e2cadd8380
1 changed files with 1 additions and 6 deletions
--- a/egs/vctk/TTS/vits/infer.py
+++ b/egs/vctk/TTS/vits/infer.py
@ -247,12 +247,7 @@ def main():
    valid_cuts = vctk.valid_cuts()
    valid_dl = vctk.valid_dataloaders(valid_cuts)

-    infer_sets: Dict = {
-        "test",
-        test_dl,
-        "valid",
-        valid_dl,
-    }
+    infer_sets = {"test": test_dl, "valid": valid_dl}

    for subset, dl in infer_sets.items():
        save_wav_dir = params.res_dir / "wav" / subset