Fix

2023-01-16 19:50:22 +08:00 · 2023-01-16 19:50:22 +08:00 · 16dd5f65f6
commit 16dd5f65f6
parent 4c5ebd2f1e
1 changed files with 1 additions and 1 deletions
--- a/vall_e/train.py
+++ b/vall_e/train.py
@ -83,7 +83,7 @@ def main():
                resps_list = model(
                    text_list=batch["text"],
                    proms_list=batch["proms"],
-                    resp_list=batch["resp"],
+                    resps_list=[r.unsqueeze(-1) for r in batch["resp"]],
                    sampling_temperature=cfg.sampling_temperature,
                )
            else: