vikarti.anatra/blt
1
0
Fork 0
mirror of https://github.com/facebookresearch/blt.git synced 2025-01-18 16:37:46 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix grad flow

Browse source
This commit is contained in:
Luciferian Ink 2025-01-18 02:02:56 -06:00
parent 5adf1c7133
commit 1ca0e04004
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
bytelatent/data/patcher.py
View file

@ -540,7 +540,7 @@ class Patcher:
if self.log_time: if self.log_time:
s = time.time() s = time.time()
if entropies is not None: if entropies is not None:
scores = entropies.clone().detach().to(dtype=torch.float32) scores = entropies.to(dtype=torch.float32)
elif preds is not None: elif preds is not None:
scores = entropy(preds) scores = entropy(preds)
else: else: