Spaces:
Running
on
Zero
Running
on
Zero
envs
Browse files
pipelines/pipeline_imagecoductor.py
CHANGED
|
@@ -410,6 +410,9 @@ class ImageConductorPipeline(DiffusionPipeline):
|
|
| 410 |
|
| 411 |
num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
|
| 412 |
|
|
|
|
|
|
|
|
|
|
| 413 |
with self.progress_bar(total=num_inference_steps) as progress_bar:
|
| 414 |
for i, t in enumerate(timesteps):
|
| 415 |
|
|
|
|
| 410 |
|
| 411 |
num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
|
| 412 |
|
| 413 |
+
self.image_controlnet.to(device)
|
| 414 |
+
self.flow_controlnet.to(device)
|
| 415 |
+
|
| 416 |
with self.progress_bar(total=num_inference_steps) as progress_bar:
|
| 417 |
for i, t in enumerate(timesteps):
|
| 418 |
|