fix KarrasVePipeline bug (#2828)

8bdf4236 · junhsss · GitHub · 585f621a · 8bdf4236
Unverified Commit 8bdf4236 authored Mar 28, 2023 by junhsss Committed by GitHub Mar 28, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py ...nes/stochastic_karras_ve/pipeline_stochastic_karras_ve.py +1 -1

No files found.
--- a/src/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py
+++ b/src/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py
@@ -120,7 +120,7 @@ class KarrasVePipeline(DiffusionPipeline):
        sample = (sample / 2 + 0.5).clamp(0, 1)
        image = sample.cpu().permute(0, 2, 3, 1).numpy()
        if output_type == "pil":
-            image = self.numpy_to_pil(sample)
+            image = self.numpy_to_pil(image)

        if not return_dict:
            return (image,)