From c92b006129bf6b91557e3558816fc8f3a601cb3b Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 10 Mar 2023 03:48:46 +0000 Subject: [PATCH] I really hate YAML --- src/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/utils.py b/src/utils.py index 24a47ee..9e1f743 100755 --- a/src/utils.py +++ b/src/utils.py @@ -1440,11 +1440,11 @@ def save_training_settings( **kwargs ): settings['learning_rate_scheme'] = "\n".join(learning_rate_schema) if settings['resume_state']: - settings['source_model'] = f"# pretrain_model_gpt: {settings['source_model']}" - settings['resume_state'] = f"resume_state: {settings['resume_state']}'" + settings['source_model'] = f"# pretrain_model_gpt: '{settings['source_model']}'" + settings['resume_state'] = f"resume_state: '{settings['resume_state']}'" else: - settings['source_model'] = f"pretrain_model_gpt: {settings['source_model']}" - settings['resume_state'] = f"# resume_state: {settings['resume_state']}'" + settings['source_model'] = f"pretrain_model_gpt: '{settings['source_model']}'" + settings['resume_state'] = f"# resume_state: '{settings['resume_state']}'" with open(f'./models/.template.yaml', 'r', encoding="utf-8") as f: yaml = f.read()