sigma_to_t -> timestep

Taurusmumu · Aug 13, 2024 · b0f9eae · b0f9eae
1 parent 1b18656
commit b0f9eae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/latent_diffusion.py b/latent_diffusion.py
@@ -858,7 +858,7 @@ def sample(self, cfg_guidance, prompt=["", ""], callback_fn=None, **kwargs):
             c_in = at.clone().sqrt()
             c_out = -sigma.clone()
 
-            new_t = self.sigma_to_t(sigma).to(self.device)
+            new_t = self.timestep(sigma).to(self.device)
 
             with torch.no_grad():
                 noise_uc, noise_c = self.predict_noise(x*c_in, new_t, uc, c)