Skip to content

Commit 1ed65f2

Browse files
committed
version bump
1 parent 6146c3c commit 1ed65f2

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

  • sub-packages/bionemo-moco

sub-packages/bionemo-moco/VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0.0.2.1
1+
0.0.2.2

sub-packages/bionemo-moco/src/bionemo/moco/interpolants/continuous_time/discrete/mdlm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -601,7 +601,7 @@ def stochastic_sample_from_categorical(self, logits: Tensor, temperature: float
601601
if temperature > 0:
602602
gumbel = -torch.log(
603603
-torch.log(torch.rand(logits.shape, device=logits.device, generator=self.rng_generator) + 1e-8) + 1e-8
604-
) #! avoid device transfers
604+
) # avoid device transfers
605605
logits = logits / temperature + noise_scale * gumbel
606606
scores, tokens = logits.log_softmax(dim=-1).max(dim=-1)
607607
return tokens, scores

0 commit comments

Comments
 (0)