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

fix ppo save model

Browse Source
This commit is contained in:
hiyouga
2023-09-12 16:25:29 +08:00
parent d4be857e23
commit 7ba57d5b14
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llmtuner/tuner/ppo/trainer.py
View File

@@ -147,7 +147,8 @@ class CustomPPOTrainer(PPOTrainer, Trainer):
dataiter = iter(self.dataloader)
steps_trained = 0
self.log_callback.on_train_end(
self.log_callback.on_train_end(self.args, self.state, self.control)
self.save_callback.on_train_end(
self.args, self.state, self.control, model=self.accelerator.unwrap_model(self.model)
)