superdiff
/

superdiff-sdxl-v1-0

SuperDiffSDXLPipeline

Model card Files Files and versions Community

mskrt commited on Mar 6

Commit

7bfed41

·

verified ·

1 Parent(s): 63aa7ec

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -443,7 +443,7 @@ class SuperDiffSDXLPipeline(DiffusionPipeline, ConfigMixin):
                     * (noise_pred_text_o - noise_pred_text_b)
                 )
-                if i < self.num_inference_steps - 1:
                     latents += 2 * dsigma * noise_pred + noise
                 else:
                     latents += dsigma * noise_pred

                     * (noise_pred_text_o - noise_pred_text_b)
                 )
+                if i < self.num_inference_steps - 3:
                     latents += 2 * dsigma * noise_pred + noise
                 else:
                     latents += dsigma * noise_pred