423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 13:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #4579

Browse Source 
Former-commit-id: 677c86594e4ea904fde0a557852daf54636b06ae
This commit is contained in:
hzhaoy 2024-06-27 13:49:57 +08:00
parent 58607ec1b0
commit e1751f6398
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/llamafactory/train/sft/trainer.py
View File

@ -53,6 +53,9 @@ class CustomSeq2SeqTrainer(Seq2SeqTrainer):
self.processor = processor
if finetuning_args.pissa_convert:
if self.is_deepspeed_enabled:
self.accelerator.deepspeed_config = self.accelerator.state.deepspeed_plugin.deepspeed_config
self.deepspeed = self._wrap_model(self.model_wrapped)
self.save_model(os.path.join(self.args.output_dir, "pissa_init"))
if finetuning_args.use_badam: