423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #4579

Browse Source 
Former-commit-id: 0fa298ff6a4febea36ea9f11c7594277a77e6e9b
This commit is contained in:
hzhaoy 2024-06-27 13:49:57 +08:00
parent d1aad72826
commit 89d9dd5aa5
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: