Fix missing variable assign in DeepFloyd-IF-II (#3315)

Fix missing variable assign lol
huggingface · May 3, 2023 · 63a8ef7 · 63a8ef7
1 parent 0ccad2a
commit 63a8ef7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py b/src/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py
@@ -667,7 +667,7 @@ def preprocess_image(self, image, num_images_per_prompt, device):
             image = [np.array(i).astype(np.float32) / 255.0 for i in image]
 
             image = np.stack(image, axis=0)  # to np
-            torch.from_numpy(image.transpose(0, 3, 1, 2))
+            image = torch.from_numpy(image.transpose(0, 3, 1, 2))
         elif isinstance(image[0], np.ndarray):
             image = np.stack(image, axis=0)  # to np
             if image.ndim == 5: