Skip to content

Commit 02d7e01

Browse files
authored
fix KarrasVePipeline bug (huggingface#2828)
1 parent c5c1c4c commit 02d7e01

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def __call__(
120120
sample = (sample / 2 + 0.5).clamp(0, 1)
121121
image = sample.cpu().permute(0, 2, 3, 1).numpy()
122122
if output_type == "pil":
123-
image = self.numpy_to_pil(sample)
123+
image = self.numpy_to_pil(image)
124124

125125
if not return_dict:
126126
return (image,)

0 commit comments

Comments
 (0)