zetavg commited on
Commit
8177d08
·
1 Parent(s): 2aa964c

write tuner version to model info

Browse files
Files changed (1) hide show
  1. llama_lora/ui/finetune_ui.py +4 -0
llama_lora/ui/finetune_ui.py CHANGED
@@ -530,6 +530,10 @@ Train data (first 10):
530
  info['continued_from_model'] = continue_from_model
531
  if continue_from_checkpoint:
532
  info['continued_from_checkpoint'] = continue_from_checkpoint
 
 
 
 
533
  json.dump(info, info_json_file, indent=2)
534
 
535
  if not should_training_progress_track_tqdm:
 
530
  info['continued_from_model'] = continue_from_model
531
  if continue_from_checkpoint:
532
  info['continued_from_checkpoint'] = continue_from_checkpoint
533
+
534
+ if Global.version:
535
+ info['tuner_version'] = Global.version
536
+
537
  json.dump(info, info_json_file, indent=2)
538
 
539
  if not should_training_progress_track_tqdm: