From 8bdf423645b80da612d821f9a0fb2977b96fe448 Mon Sep 17 00:00:00 2001 From: junhsss Date: Tue, 28 Mar 2023 23:58:19 +0900 Subject: [PATCH] fix KarrasVePipeline bug (#2828) --- .../stochastic_karras_ve/pipeline_stochastic_karras_ve.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py b/src/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py index 4535500e25..2e0ab15eb9 100644 --- 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,)