Fix er_sde steps

2025-12-29 15:36:43 +02:00
parent 19bcee67ed
commit 351829a2b6
1 changed files with 1 additions and 1 deletions
@@ -34,7 +34,7 @@ class ERSDEScheduler():
            sigmas.append(0.0)
        self.sigmas = torch.FloatTensor(sigmas)
        self.sigmas = self.shift * self.sigmas / (1 + (self.shift - 1) * self.sigmas)
-        self.timesteps = self.sigmas * self.num_train_timesteps
+        self.timesteps = self.sigmas[:-1] * self.num_train_timesteps
        self.step_index = 0
        self.old_denoised = None
        self.old_denoised_d = None