Skip to content

Commit 2d3067e

Browse files
gitmylohari10599
authored andcommitted
Fix missing variable assign in DeepFloyd-IF-II (huggingface#3315)
Fix missing variable assign lol
1 parent f7b042c commit 2d3067e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -667,7 +667,7 @@ def preprocess_image(self, image, num_images_per_prompt, device):
667667
image = [np.array(i).astype(np.float32) / 255.0 for i in image]
668668

669669
image = np.stack(image, axis=0) # to np
670-
torch.from_numpy(image.transpose(0, 3, 1, 2))
670+
image = torch.from_numpy(image.transpose(0, 3, 1, 2))
671671
elif isinstance(image[0], np.ndarray):
672672
image = np.stack(image, axis=0) # to np
673673
if image.ndim == 5:

0 commit comments

Comments
 (0)