Make save_only_last option available in UI

ddPn08 · Oct 17, 2023 · 959cc71 · 959cc71
1 parent c4a12a8
commit 959cc71
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/modules/tabs/training.py b/modules/tabs/training.py
@@ -150,6 +150,7 @@ def train_all(
             save_every_epoch,
             save_wav_with_checkpoint,
             fp16,
+            save_only_last,
             pre_trained_bottom_model_g,
             pre_trained_bottom_model_d,
             run_train_index,
@@ -264,7 +265,7 @@ def train_all(
                 pre_trained_bottom_model_d,
                 embedder_name,
                 int(embedding_output_layer),
-                False,
+                save_only_last,
                 None if len(gpu_ids) > 1 else device,
             )
 
@@ -380,6 +381,9 @@ def train_all(
                         fp16 = gr.Checkbox(
                             label="FP16", value=half_support, disabled=not half_support
                         )
+                        save_only_last = gr.Checkbox(
+                            label="Save only the latest G and D files", value=False
+                        )
                     with gr.Row(equal_height=False):
                         augment = gr.Checkbox(label="Augment", value=False)
                         augment_from_pretrain = gr.Checkbox(
@@ -477,6 +481,7 @@ def train_all(
                 save_every_epoch,
                 save_wav_with_checkpoint,
                 fp16,
+                save_only_last,
                 pre_trained_generator,
                 pre_trained_discriminator,
                 run_train_index,