fix interpolation size

pipeline.py

@@ -1185,23 +1185,25 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
             callback_steps=callback_steps,
         )
         image_tensor = torch.from_numpy(image.images[0])
-
-        # resized_image = torch.nn.functional.interpolate(
-        #     image_tensor, size=(int(height*resize_scale), int(width*resize_scale)))
-
-        # highres_image = self.__call__(prompt=prompt,
-        #                               negative_prompt=negative_prompt,
-        #                               image=resized_image,
-        #                               num_inference_steps=num_inference_steps_upscale,
-        #                               guidance_scale=guidance_scale,
-        #                               strength=strength,
-        #                               num_images_per_prompt=num_images_per_prompt,
-        #                               eta=eta,
-        #                               generator=generator,
-        #                               max_embeddings_multiples=max_embeddings_multiples,
-        #                               output_type=output_type,
-        #                               return_dict=return_dict,
-        #                               callback=callback,
-        #                               is_cancelled_callback=is_cancelled_callback,
-        #                               callback_steps=callback_steps,)
-        return image
+        image_tensor = torch.unsqueeze(
+            image_tensor.reshape([3, height, width]), 0)
+
+        resized_image = torch.nn.functional.interpolate(
+        image_tensor, size=(int(height*resize_scale), int(width*resize_scale)))
+
+        highres_image = self.__call__(prompt=prompt,
+                                      negative_prompt=negative_prompt,
+                                      image=resized_image,
+                                      num_inference_steps=num_inference_steps_upscale,
+                                      guidance_scale=guidance_scale,
+                                      strength=strength,
+                                      num_images_per_prompt=num_images_per_prompt,
+                                      eta=eta,
+                                      generator=generator,
+                                      max_embeddings_multiples=max_embeddings_multiples,
+                                      output_type=output_type,
+                                      return_dict=return_dict,
+                                      callback=callback,
+                                      is_cancelled_callback=is_cancelled_callback,
+                                      callback_steps=callback_steps,)
+        return highres_image