Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1031,6 +1031,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
             # 2) now repeat latents/images/prompts
             latents = latents.repeat_interleave(num_particles, dim=0)
             image   = image.repeat_interleave(num_particles, dim=0)
+            ram_encoder_hidden_states = ram_encoder_hidden_states.repeat_interleave(num_particles, dim=0)
             prompt_embeds = prompt_embeds.repeat_interleave(num_particles, dim=0)
             latents.requires_grad_(True)
 
@@ -1070,7 +1071,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                 if h*w<=tile_size*tile_size: # tiled latent input
                     down_block_res_samples, mid_block_res_sample = [None]*10, None
 
-                    print(f"controlnet 1 started with {controlnet_latent_model_input.shape}:{image.shape}")
+                    print(f"controlnet 1 started with {controlnet_latent_model_input.shape}:{ram_encoder_hidden_states.shape}")
                     
                     down_block_res_samples, mid_block_res_sample = self.controlnet(
                         controlnet_latent_model_input,