Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1245,7 +1245,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                     ).detach().requires_grad_(True)
                 else:
                     # compute the previous noisy sample x_t -> x_t-1
-                    latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs, return_dict=False)[0].detach()
+                    latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs, return_dict=False)[0]
 
                 # call the callback, if provided
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
@@ -1262,10 +1262,10 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
 
         has_nsfw_concept = None
         if not output_type == "latent":
-            image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]#.flip(1)
+            image = self.vae.decode(latents.detach() / self.vae.config.scaling_factor, return_dict=False)[0]#.flip(1)
             #image, has_nsfw_concept = self.run_safety_checker(image, device, prompt_embeds.dtype)
         else:
-            image = latents
+            image = latents.detach()
             has_nsfw_concept = None
 
         if has_nsfw_concept is None: