From 52aada71747a36c7e47873b3506bcf7c33e2be12 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Thu, 11 May 2023 20:22:30 +0900 Subject: [PATCH] Fix typo --- scripts/finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/finetune.py b/scripts/finetune.py index ff2f87eb1..d6a920f5d 100644 --- a/scripts/finetune.py +++ b/scripts/finetune.py @@ -236,7 +236,7 @@ def train( logging.info(f"Training Completed!!! Saving pre-trained model to {cfg.output_dir}") # TODO do we need this fix? https://huggingface.co/docs/accelerate/usage_guides/fsdp#saving-and-loading - trainer.save_pretrained(cfg.output_dir) + model.save_pretrained(cfg.output_dir) # trainer.save_model(cfg.output_dir) # TODO this may be needed for deepspeed to work? need to review another time