From 17e937aae9c90b513e4b7674f442a60161c84e83 Mon Sep 17 00:00:00 2001
From: Lordmau5 <mail@lordmau5.com>
Date: Thu, 13 Apr 2023 13:00:48 +0200
Subject: [PATCH] feat(train): use RichProgressBar (#311)

---
 src/so_vits_svc_fork/train.py | 1 +
 1 file changed, 1 insertion(+)

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)