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 |
|