diff --git a/bytelatent/data/patcher.py b/bytelatent/data/patcher.py
index b495d9b..7e0396a 100644
--- a/bytelatent/data/patcher.py
+++ b/bytelatent/data/patcher.py
@@ -540,7 +540,7 @@ class Patcher:
             if self.log_time:
                 s = time.time()
             if entropies is not None:
-                scores = entropies.clone().detach().to(dtype=torch.float32)
+                scores = entropies.to(dtype=torch.float32)
             elif preds is not None:
                 scores = entropy(preds)
             else: