envs

pipelines/pipeline_imagecoductor.py

@@ -402,6 +402,7 @@ class ImageConductorPipeline(DiffusionPipeline):
         obj_latents = copy.deepcopy(latents)
         cam_latents = copy.deepcopy(latents)
         
+        print("device", device)
         # Prepare extra step kwargs.
         extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
 
@@ -457,7 +458,13 @@ class ImageConductorPipeline(DiffusionPipeline):
                         controlnet_images_mask = controlnet_images_mask.half()
                         controlnet_flows = controlnet_flows.half()
                         text_embeddings = text_embeddings.half()
-
+                    print("controlnet_noisy_latents device", controlnet_noisy_latents.device)
+                    print("controlnet_prompt_embeds device", controlnet_prompt_embeds.device)
+                    print("controlnet_images device", controlnet_images.device)
+                    
+                    
+                    
+                    
                     img_down_block_additional_residuals, img_mid_block_additional_residuals = self.image_controlnet(
                                 controlnet_noisy_latents, t,
                                 encoder_hidden_states=controlnet_prompt_embeds,