Spaces:
Runtime error
Runtime error
Update pipelines/pipeline_seesr.py
Browse files
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).detach()
|
| 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].detach()
|
| 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):
|