diff --git a/pipelines/deepfloyd_if/pipeline_if_superresolution.py b/pipelines/deepfloyd_if/pipeline_if_superresolution.py index 4729cec3e4d7..1ba8f888a8e3 100644 --- a/pipelines/deepfloyd_if/pipeline_if_superresolution.py +++ b/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: