rm print(self.model) in trainer.py for reranker

FlagOpen · Nov 15, 2024 · 6c448d7 · 6c448d7
1 parent a08f352
commit 6c448d7
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/FlagEmbedding/finetune/reranker/encoder_only/base/trainer.py b/FlagEmbedding/finetune/reranker/encoder_only/base/trainer.py
@@ -29,7 +29,6 @@ def _save(self, output_dir: Optional[str] = None, state_dict=None):
         if not hasattr(self.model, 'save_pretrained'):
             raise NotImplementedError(f'MODEL {self.model.__class__.__name__} ' f'does not support save_pretrained interface')
         else:
-            print(self.model)
             self.model.save_pretrained(output_dir)
         if self.tokenizer is not None and self.is_world_process_zero():
             self.tokenizer.save_pretrained(output_dir)