controlnet more gpt fix
Browse files- pipeline.py +2 -1
pipeline.py
CHANGED
|
@@ -1013,11 +1013,12 @@ class StableDiffusionControlNetImg2ImgPipeline(DiffusionPipeline, TextualInversi
|
|
| 1013 |
down_block_res_sample, mid_block_res_sample = self.controlnet(
|
| 1014 |
latent_model_input[i * num_images_per_prompt:(i + 1) * num_images_per_prompt],
|
| 1015 |
t,
|
| 1016 |
-
encoder_hidden_states=prompt_embeds,
|
| 1017 |
controlnet_cond=controlnet_conditioning_image[i],
|
| 1018 |
conditioning_scale=controlnet_conditioning_scale,
|
| 1019 |
return_dict=False,
|
| 1020 |
)
|
|
|
|
| 1021 |
down_block_res_samples.append(down_block_res_sample)
|
| 1022 |
mid_block_res_samples.append(mid_block_res_sample)
|
| 1023 |
|
|
|
|
| 1013 |
down_block_res_sample, mid_block_res_sample = self.controlnet(
|
| 1014 |
latent_model_input[i * num_images_per_prompt:(i + 1) * num_images_per_prompt],
|
| 1015 |
t,
|
| 1016 |
+
encoder_hidden_states=prompt_embeds[i * num_images_per_prompt:(i + 1) * num_images_per_prompt],
|
| 1017 |
controlnet_cond=controlnet_conditioning_image[i],
|
| 1018 |
conditioning_scale=controlnet_conditioning_scale,
|
| 1019 |
return_dict=False,
|
| 1020 |
)
|
| 1021 |
+
|
| 1022 |
down_block_res_samples.append(down_block_res_sample)
|
| 1023 |
mid_block_res_samples.append(mid_block_res_sample)
|
| 1024 |
|