Updated for removing use_tqdm=False

castorini · Sep 25, 2024 · 406fca7 · 406fca7
1 parent 36beed1
commit 406fca7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rank_llm/rerank/listwise/rank_listwise_os_llm.py b/src/rank_llm/rerank/listwise/rank_listwise_os_llm.py
@@ -171,7 +171,7 @@ def run_llm(
         # gen_cfg.temperature = 0
         gen_cfg.do_sample = False
         output_ids = self._llm.generate(
-            **inputs, use_tqdm=not silence, generation_config=gen_cfg
+            **inputs, generation_config=gen_cfg
         )
 
         if self._llm.config.is_encoder_decoder: