yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2026-01-06 12:11:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
hanhainebula 2024-11-15 15:08:59 +08:00
parent a08f3524c5
commit 6c448d7065
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
FlagEmbedding/finetune/reranker/encoder_only/base/trainer.py
View File

@ -29,7 +29,6 @@ class EncoderOnlyRerankerTrainer(AbsRerankerTrainer):
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)