oops

2024-06-03 22:35:55 -05:00 · 2024-06-03 22:35:55 -05:00 · 186b93a77e
commit 186b93a77e
parent e50edc3b48
1 changed files with 1 additions and 1 deletions
--- a/vall_e/train.py
+++ b/vall_e/train.py
@ -137,7 +137,7 @@ def run_eval(engines, eval_name, dl):
 def train():
 	parser = argparse.ArgumentParser("VALL-E TTS")
 	parser.add_argument("--eval", action="store_true", default=None)
-	args = parser.parse_args()
+	args, unknown = parser.parse_known_args()

 	setup_logging(cfg.log_dir)