diff --git a/opencompass/openicl/icl_inferencer/cot_inferencer.py b/opencompass/openicl/icl_inferencer/cot_inferencer.py index 46084a3..52ad499 100644 --- a/opencompass/openicl/icl_inferencer/cot_inferencer.py +++ b/opencompass/openicl/icl_inferencer/cot_inferencer.py @@ -163,7 +163,7 @@ def inference(self, sc_thoughts.append(thoughts) sc_results.append(results) sc_prediction = list(map(list, zip(*sc_results))) - # sc_thoughts = list(map(list, zip(*sc_thoughts))) + sc_thoughts = list(map(list, zip(*sc_thoughts))) generated = sc_prediction # 5-3. Save current output