Skip to content

Commit ccb2b89

Browse files
authored
Revert incorrect line in deterministic sampler (#974)
* Revert incorrect line in deterministic sampler * minor version bump
1 parent 7a798a3 commit ccb2b89

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

physicsnemo/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,4 +19,4 @@
1919
from .models.meta import ModelMetaData
2020
from .models.module import Module
2121

22-
__version__ = "1.1.0"
22+
__version__ = "1.1.1"

physicsnemo/utils/generative/deterministic_sampler.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -334,8 +334,7 @@ def deterministic_sampler(
334334
).to(torch.float64)
335335
d_prime = (
336336
sigma_deriv(t_prime) / sigma(t_prime) + s_deriv(t_prime) / s(t_prime)
337-
) * x_prime
338-
-sigma_deriv(t_prime) * s(t_prime) / sigma(t_prime) * denoised
337+
) * x_prime - sigma_deriv(t_prime) * s(t_prime) / sigma(t_prime) * denoised
339338
x_next = x_hat + h * (
340339
(1 - 1 / (2 * alpha)) * d_cur + 1 / (2 * alpha) * d_prime
341340
)

0 commit comments

Comments
 (0)