Update layerwise.py

FlagOpen · Dec 3, 2024 · 4e636fe · 4e636fe
1 parent 777a9ac
commit 4e636fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/FlagEmbedding/inference/reranker/decoder_only/layerwise.py b/FlagEmbedding/inference/reranker/decoder_only/layerwise.py
@@ -374,7 +374,7 @@ def compute_score_single_gpu(
             if normalize:
                 all_scores[i] = [sigmoid(score) for score in all_scores[i]]
 
-        if isinstance(all_scores[0], list):
+        if len(all_scores) == 1 and isinstance(all_scores[0], list):
             all_scores = all_scores[0]
 
         return all_scores