From 83528219844d299b5bfe3a0caaa8cb945a2ffbb7 Mon Sep 17 00:00:00 2001 From: TyrianOtter <154254636+TyrianOtter@users.noreply.github.com> Date: Mon, 30 Sep 2024 04:20:29 -0400 Subject: [PATCH] fix `DPMSolver.__call__` docs calling it deterministic (#452) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com> --- src/refiners/foundationals/latent_diffusion/solvers/dpm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/refiners/foundationals/latent_diffusion/solvers/dpm.py b/src/refiners/foundationals/latent_diffusion/solvers/dpm.py index 0ff2ed028..2ec2de446 100644 --- a/src/refiners/foundationals/latent_diffusion/solvers/dpm.py +++ b/src/refiners/foundationals/latent_diffusion/solvers/dpm.py @@ -293,7 +293,7 @@ def __call__( x: The input tensor to apply the diffusion process to. predicted_noise: The predicted noise tensor for the current step. step: The current step of the diffusion process. - generator: The random number generator to use for sampling noise (ignored, this solver is deterministic). + generator: The random number generator to use for sampling noise (ignored, except if `sde_variance=1.0` in CTOR params) Returns: The denoised version of the input data `x`.