deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-05 18:29:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

tiny fix

Browse Source
This commit is contained in:
fzc8578
2025-01-11 13:58:19 +08:00
parent c5e82b1bc7
commit 4a087a8aec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/trainer.py
View File

@@ -246,7 +246,7 @@ class CPMTrainer(Trainer):
if self.tokenizer is not None: if self.tokenizer is not None:
self.tokenizer.save_pretrained(output_dir) self.tokenizer.save_pretrained(output_dir)
if getattr(self.model, "processor") is not None: if self.model.processor is not None:
self.model.processor.save_pretrained(output_dir) self.model.processor.save_pretrained(output_dir)
# Good practice: save your training arguments together with the trained model # Good practice: save your training arguments together with the trained model