diff --git a/src/so_vits_svc_fork/train.py b/src/so_vits_svc_fork/train.py index 9c90af39..4475cf74 100644 --- a/src/so_vits_svc_fork/train.py +++ b/src/so_vits_svc_fork/train.py @@ -82,6 +82,7 @@ def train( if hparams.train.get("bf16_run", False) else 32, strategy=strategy, + callbacks=[pl.callbacks.RichProgressBar()], ) model = VitsLightning(reset_optimizer=reset_optimizer, **hparams) trainer.fit(model, datamodule=datamodule)