From 113966157cc5163aee909424a2515274534c5058 Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Thu, 5 Sep 2024 15:54:33 +0800
Subject: [PATCH] fix: fix max steps

---
 prepare_yaml_file.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index 36357f6b..fb27f279 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -43,7 +43,7 @@ def main():
     config['template'] = template
     config['output_dir'] = output_dir
     if run_type == "lora_sft":
-        config['max_steps'] = max_steps
+        config['max_steps'] = int(max_steps)
 
     with open(f'{output_dir}/{run_name}.yaml', 'w', encoding='utf-8') as f:
         yaml.dump(data=config, stream=f, allow_unicode=True)