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