From 91d1b46aa963ec07651b3541625f5cba1e4dffa1 Mon Sep 17 00:00:00 2001 From: Pierre Chapuis Date: Mon, 26 Feb 2024 10:47:56 +0100 Subject: [PATCH] Add a note that we mitigate non-zero SNR in DDIM. --- src/refiners/foundationals/latent_diffusion/solvers/solver.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/refiners/foundationals/latent_diffusion/solvers/solver.py b/src/refiners/foundationals/latent_diffusion/solvers/solver.py index ad9601c..caaaaec 100644 --- a/src/refiners/foundationals/latent_diffusion/solvers/solver.py +++ b/src/refiners/foundationals/latent_diffusion/solvers/solver.py @@ -33,7 +33,8 @@ class TimestepSpacing(str, Enum): Attributes: LINSPACE: Sample N steps with linear interpolation, return a floating-point tensor. LINSPACE_ROUNDED: Same as LINSPACE but return an integer tensor with rounded timesteps. - LEADING: Sample N+1 steps, do not include the last timestep (i.e. bad - non-zero SNR). Used in DDIM. + LEADING: Sample N+1 steps, do not include the last timestep (i.e. bad - non-zero SNR). + Used in DDIM, with a mitigation for that issue. TRAILING: Sample N+1 steps, do not include the first timestep. CUSTOM: Use custom timespacing in solver (override `_generate_timesteps`, see DPM). """