fix: Add logprobs to generator generate method in SelfRAGPipeline

RUC-NLPIR · Nov 22, 2024 · 8578980 · 8578980
1 parent 2bf75ee
commit 8578980
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flashrag/pipeline/active_pipeline.py b/flashrag/pipeline/active_pipeline.py
@@ -417,7 +417,7 @@ def run_single_beam(self, prompt, item_retrieval_result=None):
                         else:
                             aug_prompts = [prompt + prev_generation]
 
-                        item_pred = self.generator.generate(aug_prompts, return_raw_output=True)
+                        item_pred = self.generator.generate(aug_prompts, return_raw_output=True, logprobs=5)
                         _, preds, scores, overall_score_dict = self.critic_preds(item_pred)
 
                         for i, (pred, p_score) in enumerate(zip(preds, scores)):