diff --git a/scripts/finetune.py b/scripts/finetune.py index 1da332f40..a7c3d6c77 100644 --- a/scripts/finetune.py +++ b/scripts/finetune.py @@ -230,6 +230,9 @@ def train( ) # TODO do we need this fix? https://huggingface.co/docs/accelerate/usage_guides/fsdp#saving-and-loading trainer.save_model(cfg.output_dir) + + if cfg.adapter == 'lora': + trainer.save_pretrained(cfg.output_dir) if __name__ == "__main__":