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

I really hate YAML

Browse Source
This commit is contained in:
mrq 2023-03-10 03:48:46 +00:00
parent d3184004fd
commit c92b006129
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/utils.py
View File

@ -1440,11 +1440,11 @@ def save_training_settings( **kwargs ):
settings['learning_rate_scheme'] = "\n".join(learning_rate_schema) settings['learning_rate_scheme'] = "\n".join(learning_rate_schema)
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']}'"
settings['resume_state'] = f"resume_state: {settings['resume_state']}'" settings['resume_state'] = f"resume_state: '{settings['resume_state']}'"
else: else:
settings['source_model'] = f"pretrain_model_gpt: {settings['source_model']}" settings['source_model'] = f"pretrain_model_gpt: '{settings['source_model']}'"
settings['resume_state'] = f"# resume_state: {settings['resume_state']}'" settings['resume_state'] = f"# resume_state: '{settings['resume_state']}'"
with open(f'./models/.template.yaml', 'r', encoding="utf-8") as f: with open(f'./models/.template.yaml', 'r', encoding="utf-8") as f:
yaml = f.read() yaml = f.read()