envs

pipelines/pipeline_imagecoductor.py

@@ -410,6 +410,9 @@ class ImageConductorPipeline(DiffusionPipeline):
 
         num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
 
+        self.image_controlnet.to(device)
+        self.flow_controlnet.to(device)
+        
         with self.progress_bar(total=num_inference_steps) as progress_bar:
             for i, t in enumerate(timesteps):