alexnasa commited on
Commit
9fdf514
·
verified ·
1 Parent(s): 63f12be

Update pipelines/pipeline_seesr.py

Browse files
Files changed (1) hide show
  1. pipelines/pipeline_seesr.py +1 -1
pipelines/pipeline_seesr.py CHANGED
@@ -1245,7 +1245,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
1245
  ).detach().requires_grad_(True)
1246
  else:
1247
  # compute the previous noisy sample x_t -> x_t-1
1248
- latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs, return_dict=False)[0].requires_grad_(False)
1249
 
1250
  # call the callback, if provided
1251
  if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
 
1245
  ).detach().requires_grad_(True)
1246
  else:
1247
  # compute the previous noisy sample x_t -> x_t-1
1248
+ latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs, return_dict=False)[0]
1249
 
1250
  # call the callback, if provided
1251
  if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):