Spaces:
Running
on
Zero
Running
on
Zero
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)[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):
|
|
|
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):
|