Yw22 commited on
Commit
f935254
1 Parent(s): 45f3350
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