AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

fix KarrasVePipeline bug (#2828)

Browse Source
This commit is contained in:
junhsss
2023-03-28 23:58:19 +09:00
committed by GitHub
parent 585f621af2
commit 8bdf423645
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py
View File

@@ -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,)