update save_model

DesmonDay · Oct 17, 2024 · 27bb12f · 27bb12f
1 parent b2639ac
commit 27bb12f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/paddlenlp/trainer/trainer.py b/paddlenlp/trainer/trainer.py
@@ -2372,7 +2372,11 @@ def _save_checkpoint(self, model, metrics=None):
         elif isinstance(self.model, LoRAModel) or isinstance(self.model, PrefixModelForCausalLM):
             self.save_model(output_dir, True, signal_dir)
         else:
-            self.save_model(output_dir, False, signal_dir)
+            signature = inspect.signature(self.save_model)
+            if "signal_dir" in signature.parameters:
+                self.save_model(output_dir, False, signal_dir)
+            else:
+                self.save_model(output_dir)
 
         # only save model state dict, ignore optimizer and scheduler
         if not self.args.ignore_save_lr_and_optim: