Remove reference to score

pomonam · Jul 8, 2024 · 8f5e80f · 8f5e80f
1 parent 7556c56
commit 8f5e80f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kronfluence/score/dot_product.py b/kronfluence/score/dot_product.py
@@ -114,7 +114,8 @@ def compute_dot_products_with_loader(
                             if score_args.compute_per_token_scores:
                                 raise RuntimeError(DIMENSION_NOT_MATCH_ERROR_MSG) from exc
                             raise
-                    score_chunks[ALL_MODULE_NAME].append(pairwise_scores.cpu())
+                    pairwise_scores = pairwise_scores.cpu()
+                    score_chunks[ALL_MODULE_NAME].append(pairwise_scores)
                     accumulate_iterations(model=model, tracked_module_names=tracked_module_names)
 
             if state.use_distributed and total_steps % DISTRIBUTED_SYNC_INTERVAL == 0: