From 4a087a8aec739dee07f7e4f8fb75a505a5c75725 Mon Sep 17 00:00:00 2001
From: fzc8578 <1428195643@qq.com>
Date: Sat, 11 Jan 2025 13:58:19 +0800
Subject: [PATCH] tiny fix

---
 finetune/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/finetune/trainer.py b/finetune/trainer.py
index e1212ff..3443e34 100644
--- a/finetune/trainer.py
+++ b/finetune/trainer.py
@@ -246,7 +246,7 @@ class CPMTrainer(Trainer):
         if self.tokenizer is not None:
             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)
 
         # Good practice: save your training arguments together with the trained model