Update pipeline.py

pipeline.py

@@ -915,7 +915,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
         # image = self.decode_latents(upscale_latents)
         
         if return_latents:
-            return latents
+            return latents,noise,init_latents_orig
         else:
             image = self.decode_latents(latents)
 
@@ -1011,7 +1011,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
             (nsfw) content, according to the `safety_checker`.
         """
 
-        latents = self.__call__(
+        latents,noise,init_latents_orig = self.__call__(
             prompt=prompt,
             negative_prompt=negative_prompt,
             height=height,
@@ -1032,19 +1032,21 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
         )
 
         latents = torch.nn.functional.interpolate(latents, size=(int(height*resize_scale)//8, int(width*resize_scale)//8))
-
+        print("noise")
+        print(noise)
+        
         return self.__call__(
             prompt=prompt,
             negative_prompt=negative_prompt,
-            height=height*resize_scale,
-            width=width*resize_scale,
+            height=height,
+            width=width,
             num_inference_steps=num_inference_steps,
             guidance_scale=guidance_scale,
             strength=strength,
             num_images_per_prompt=num_images_per_prompt,
             eta=eta,
             generator=generator,
-            latents=latents,
+            latents=None,
             return_latents=False,
             max_embeddings_multiples=max_embeddings_multiples,
             output_type=output_type,