Merge branch 'main' of https://github.com/BatsResearch/alfred

BatsResearch · Jun 11, 2024 · dd837cc · dd837cc
2 parents 2faeef2 + 39b1bb2
commit dd837cc
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["logit"]
+            scores[response_idx] = response["logits"]
             candidates.append(response["candidate"])
 
         logits = {