update trainer

Author: 545999961

FlagEmbedding/abc/finetune/embedder/AbsTrainer.py

@@ -11,7 +11,7 @@ class AbsEmbedderTrainer(ABC, Trainer):
     def _save(self, output_dir: Optional[str] = None, state_dict=None):
         pass
 
-    def compute_loss(self, model, inputs, return_outputs=False):
+    def compute_loss(self, model, inputs, return_outputs=False, **kwargs):
         """
         How the loss is computed by Trainer. By default, all models return the loss in the first element.
 

FlagEmbedding/abc/finetune/reranker/AbsTrainer.py

@@ -11,7 +11,7 @@ class AbsRerankerTrainer(ABC, Trainer):
     def _save(self, output_dir: Optional[str] = None, state_dict=None):
         pass
 
-    def compute_loss(self, model, inputs, return_outputs=False):
+    def compute_loss(self, model, inputs, return_outputs=False, **kwargs):
         """
         How the loss is computed by Trainer. By default, all models return the loss in the first element.