Update gradio demo

coqui-ai · Nov 24, 2023 · c76fb85 · c76fb85
1 parent 8967fc7
commit c76fb85
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/TTS/demos/xtts_ft_demo/utils/gpt_train.py b/TTS/demos/xtts_ft_demo/utils/gpt_train.py
@@ -23,7 +23,7 @@ def train_gpt(language, num_epochs, batch_size, train_csv, eval_csv, output_path
     OPTIMIZER_WD_ONLY_ON_WEIGHTS = True  # for multi-gpu training please make it False
     START_WITH_EVAL = True  # if True it will star with evaluation
     BATCH_SIZE = batch_size  # set here the batch size
-    GRAD_ACUMM_STEPS = 1  # set here the grad accumulation steps
+    GRAD_ACUMM_STEPS = 4  # set here the grad accumulation steps
     # Note: we recommend that BATCH_SIZE * GRAD_ACUMM_STEPS need to be at least 252 for more efficient training. You can increase/decrease BATCH_SIZE but then set GRAD_ACUMM_STEPS accordingly.
 
 

diff --git a/TTS/demos/xtts_ft_demo/xtts_demo.py b/TTS/demos/xtts_ft_demo/xtts_demo.py
@@ -189,7 +189,7 @@ def preprocess_dataset(audio_path, language, out_path, progress=gr.Progress(trac
             minimum=2,
             maximum=512,
             step=1,
-            value=16,
+            value=4,
         )
         progress_train = gr.Label(
             label="Progress:"