diff --git a/llama_lora/ui/finetune_ui.py b/llama_lora/ui/finetune_ui.py index 1ce51a1..54e14c2 100644 --- 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: