Merge branch 'new-flagembedding-v1' of github.com:hanhainebula/FlagEmbedding into new-flagembedding-v1

Author: hanhainebula

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.
 

setup.py

@@ -15,7 +15,7 @@
     include_package_data=True,
     install_requires=[
         'torch>=1.6.0',
-        'transformers>=4.33.0',
+        'transformers==4.45.2',
         'datasets>=2.19.0',
         'accelerate>=0.20.1',
         'sentence_transformers',