write tuner version to model info

zetavg · Apr 21, 2023 · 8177d08 · 8177d08
1 parent 2aa964c
commit 8177d08
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/llama_lora/ui/finetune_ui.py b/llama_lora/ui/finetune_ui.py
@@ -530,6 +530,10 @@ def on_step_end(self, args, state, control, **kwargs):
                 info['continued_from_model'] = continue_from_model
                 if continue_from_checkpoint:
                     info['continued_from_checkpoint'] = continue_from_checkpoint
+
+            if Global.version:
+                info['tuner_version'] = Global.version
+
             json.dump(info, info_json_file, indent=2)
 
         if not should_training_progress_track_tqdm: