style

Author: natolambert

src/diffusers/schedulers/scheduling_ddim.py

@@ -278,7 +278,6 @@ def step(
             # need to merge the PRs for sigma to be available in DDPM
             pred_original_sample = sample * self.alphas[timestep] - model_output * self.sigmas[timestep]
             eps = model_output * self.alphas[timestep] - sample * self.sigmas[timestep]
-            raise NotImplementedError(f"v prediction not yet implemented for DDPM")
         else:
             raise ValueError(f"prediction_type given as {prediction_type} must be one of `epsilon`, `sample`, or `v`")
 

src/diffusers/schedulers/scheduling_ddpm.py

@@ -269,7 +269,6 @@ def step(
             # need to merge the PRs for sigma to be available in DDPM
             pred = sample * self.alphas[timestep] - model_output * self.sigmas[timestep]
             eps = model_output * self.alphas[timestep] - sample * self.sigmas[timestep]
-            raise NotImplementedError(f"v prediction not yet implemented for DDPM")
         else:
             raise ValueError(f"prediction_type given as {prediction_type} must be one of `epsilon`, `sample`, or `v`")