Merge pull request #69 from BatsResearch/patch-utils

Update utils.py
BatsResearch · Jun 11, 2024 · 00c5fdb · 00c5fdb
2 parents dd837cc + f9d2ec4
commit 00c5fdb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alfred/fm/utils.py b/alfred/fm/utils.py
@@ -382,7 +382,7 @@ def merge_rank_response(
         scores = torch.empty(self.candidate_size)
         candidates = []
         for response_idx, response in enumerate(responses):
-            scores[response_idx] = response["logits"]
+            scores[response_idx] = response["logit"]
             candidates.append(response["candidate"])
 
         logits = {