Skip to content

Commit 1f87d62

Browse files
authored
Fix missing variable assign in DeepFloyd-IF-II (huggingface#3315)
Fix missing variable assign lol
1 parent cd42a2e commit 1f87d62

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pipelines/deepfloyd_if/pipeline_if_superresolution.py

Lines changed: 1 addition & 1 deletion
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)