diff --git a/finetune/finetune.py b/finetune/finetune.py index 2cbd6cc..9b0ce1b 100644 --- a/finetune/finetune.py +++ b/finetune/finetune.py @@ -190,3 +190,5 @@ if __name__ == "__main__": ) trainer.train() + # save the incremental PEFT weights, more details can be found in https://huggingface.co/blog/peft + # model.save_pretrained("output_dir")