Update pipeline.py

pipeline.py

@@ -587,7 +587,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
             else:
                 noise = torch.randn(shape, generator=generator, device=device, dtype=dtype)
             latents = self.scheduler.add_noise(init_latents, noise, timestep)
-        return latents, init_latents_orig, noise
+            return latents, init_latents_orig, noise
 
     @torch.no_grad()
     def __call__(