diff --git a/vall_e/train.py b/vall_e/train.py
index 7b2ea7c..a9339e8 100755
--- a/vall_e/train.py
+++ b/vall_e/train.py
@@ -155,9 +155,11 @@ def run_eval(engines, eval_name, dl):
 
 
 def train():
+	"""
 	parser = argparse.ArgumentParser("VALL-E TTS")
 	parser.add_argument("--eval", action="store_true")
 	args = parser.parse_args()
+	"""
 
 	setup_logging(cfg.log_dir)
 
@@ -180,8 +182,10 @@ def train():
 
 	qnt.unload_model()
 
+	"""
 	if args.eval:
 		return eval_fn(engines=trainer.load_engines())
+	"""
 
 	"""
 	if cfg.trainer.load_webui: