Allow overwrite

app.py

@@ -316,10 +316,10 @@ def run_context_length_probing(
         logprobs = torch.cat([unigram_logprobs.unsqueeze(0), logprobs], dim=0)
 
         if metric == "NLL loss":
-            scores = nll_score(logprobs=logprobs, labels=label_ids)
+            scores = nll_score(logprobs=logprobs, labels=label_ids, allow_overwrite=True)
         elif metric == "KL divergence":
-            scores = kl_div_score(logprobs, labels=label_ids)
-        del logprobs  # possibly destroyed by the score computation to save memory
+            scores = kl_div_score(logprobs, labels=label_ids, allow_overwrite=True)
+        del logprobs  # possibly overwritten by the score computation to save memory
 
         scores = (-scores).diff(dim=0).transpose(0, 1)
         scores = scores.nan_to_num()