From c47f710b84c1ca5c1b452e0218ce2065dd4d778e Mon Sep 17 00:00:00 2001 From: SillyXu Date: Thu, 8 Feb 2024 10:41:37 +0900 Subject: [PATCH] Update finetune.py --- finetune/finetune.py | 2 ++ 1 file changed, 2 insertions(+) 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")