diff --git a/src/so_vits_svc_fork/train.py b/src/so_vits_svc_fork/train.py index 683b188c..1dce3faa 100644 --- a/src/so_vits_svc_fork/train.py +++ b/src/so_vits_svc_fork/train.py @@ -88,7 +88,7 @@ def train( if hparams.train.get("bf16_run", False) else 32, strategy=strategy, - callbacks=[pl.callbacks.RichProgressBar()] if is_notebook() else None, + callbacks=[pl.callbacks.RichProgressBar()] if not is_notebook() else None, ) model = VitsLightning(reset_optimizer=reset_optimizer, **hparams) trainer.fit(model, datamodule=datamodule)