mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 223 Pull Requests 5 Packages Projects Releases Wiki Activity

forgot to save the simplified training input json first before touching any of the settings that dump to the yaml

Browse Source
This commit is contained in:
mrq 2023-03-09 02:27:20 +00:00
parent 221ac38b32
commit 1b18b3e335
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/utils.py
View File

@ -1348,6 +1348,10 @@ def save_training_settings( **kwargs ):
settings = {} settings = {}
settings.update(kwargs) settings.update(kwargs)
outjson = f'./training/{settings["voice"]}/train.json'
with open(outjson, 'w', encoding="utf-8") as f:
f.write(json.dumps(settings, indent='\t') )
settings['dataset_path'] = f"./training/{settings['voice']}/train.txt" settings['dataset_path'] = f"./training/{settings['voice']}/train.txt"
settings['validation_path'] = f"./training/{settings['voice']}/validation.txt" settings['validation_path'] = f"./training/{settings['voice']}/validation.txt"
@ -1428,9 +1432,6 @@ def save_training_settings( **kwargs ):
if settings['dataset_path'] == settings['validation_path'] or not os.path.exists(settings['validation_path']): if settings['dataset_path'] == settings['validation_path'] or not os.path.exists(settings['validation_path']):
settings['validation_enabled'] = 'false' settings['validation_enabled'] = 'false'
""" """
outjson = f'./training/{settings["voice"]}/train.json'
with open(outjson, 'w', encoding="utf-8") as f:
f.write(json.dumps(settings, indent='\t') )
if settings['resume_state']: if settings['resume_state']:
settings['source_model'] = f"# pretrain_model_gpt: {settings['source_model']}" settings['source_model'] = f"# pretrain_model_gpt: {settings['source_model']}"

2
src/webui.py
View File

@ -414,7 +414,7 @@ def setup_gradio():
TRAINING_SETTINGS["gpus"] = gr.Number(label="GPUs", value=get_device_count(), 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["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") TRAINING_SETTINGS["resume_state"] = gr.Textbox(label="Resume State Path", placeholder="./training/${voice}/finetune/training_state/${last_state}.state")
TRAINING_SETTINGS["voice"] = gr.Dropdown( choices=dataset_list, label="Dataset", type="value", value=dataset_list[0] if len(dataset_list) else "" ) TRAINING_SETTINGS["voice"] = gr.Dropdown( choices=dataset_list, label="Dataset", type="value", value=dataset_list[0] if len(dataset_list) else "" )