From 221ac38b32c6f18b367456e28038162b91ee45d2 Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 9 Mar 2023 02:25:32 +0000 Subject: [PATCH] forgot to update to finetune subdir --- src/webui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/webui.py b/src/webui.py index 18831b5..7db6126 100755 --- a/src/webui.py +++ b/src/webui.py @@ -198,7 +198,7 @@ def optimize_training_settings_proxy( *args ): def import_training_settings_proxy( voice ): messages = [] injson = f'./training/{voice}/train.json' - statedir = f'./training/{voice}/training_state/' + statedir = f'./training/{voice}/finetune/training_state/' with open(injson, 'r', encoding="utf-8") as f: settings = json.loads(f.read()) @@ -412,7 +412,7 @@ def setup_gradio(): with gr.Row(): TRAINING_SETTINGS["workers"] = gr.Number(label="Worker Processes", value=2, precision=0) TRAINING_SETTINGS["gpus"] = gr.Number(label="GPUs", value=get_device_count(), precision=0) - + TRAINING_SETTINGS["source_model"] = gr.Dropdown( choices=autoregressive_models, label="Source Model", type="value", value=autoregressive_models[0] ) TRAINING_SETTINGS["resume_state"] = gr.Textbox(label="Resume State Path", placeholder="./training/${voice}/training_state/${last_state}.state")