Update flag_reranker.py

Author: 545999961

FlagEmbedding/flag_reranker.py

@@ -282,14 +282,14 @@ def compute_score(self, sentence_pairs: Union[List[Tuple[str, str]], Tuple[str,
         if use_dataloader:
             if num_workers is None:
                 num_workers = min(batch_size, 16)
-                dataset = DatasetForReranker(sentences_sorted,
-                                             self.model_name_or_path,
-                                             max_length,
-                                             cache_dir=self.cache_dir,
-                                             prompt=prompt)
-                dataloader = DataLoader(dataset, shuffle=False, batch_size=batch_size, drop_last=False,
-                                        num_workers=num_workers,
-                                        collate_fn=collater(self.tokenizer, max_length))
+            dataset = DatasetForReranker(sentences_sorted,
+                                         self.model_name_or_path,
+                                         max_length,
+                                         cache_dir=self.cache_dir,
+                                         prompt=prompt)
+            dataloader = DataLoader(dataset, shuffle=False, batch_size=batch_size, drop_last=False,
+                                    num_workers=num_workers,
+                                    collate_fn=collater(self.tokenizer, max_length))
 
         all_scores = []
         if dataloader is not None:
@@ -456,14 +456,14 @@ def compute_score(self, sentence_pairs: Union[List[Tuple[str, str]], Tuple[str,
         if use_dataloader:
             if num_workers is None:
                 num_workers = min(batch_size, 16)
-                dataset = DatasetForReranker(sentences_sorted,
-                                             self.model_name_or_path,
-                                             max_length,
-                                             cache_dir=self.cache_dir,
-                                             prompt=prompt)
-                dataloader = DataLoader(dataset, shuffle=False, batch_size=batch_size, drop_last=False,
-                                        num_workers=num_workers,
-                                        collate_fn=collater(self.tokenizer, max_length))
+            dataset = DatasetForReranker(sentences_sorted,
+                                         self.model_name_or_path,
+                                         max_length,
+                                         cache_dir=self.cache_dir,
+                                         prompt=prompt)
+            dataloader = DataLoader(dataset, shuffle=False, batch_size=batch_size, drop_last=False,
+                                    num_workers=num_workers,
+                                    collate_fn=collater(self.tokenizer, max_length))
 
         all_scores = []
         if dataloader is not None:
@@ -577,4 +577,4 @@ def _text_length(self, text: Union[List[int], List[List[int]]]):
         elif len(text) == 0 or isinstance(text[0], int):  # Empty string or list of ints
             return len(text)
         else:
-            return sum([len(t) for t in text])  # Sum of length of individual strings
+            return sum([len(t) for t in text])  # Sum of length of individual strings