Fix wrong conditioning used

This commit is contained in:
Sergey Borisov 2023-06-27 01:18:15 +03:00
parent 3c30368c62
commit dc1f220b3e

View File

@ -635,7 +635,7 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
control_latent_input = torch.cat([unet_latent_input] * 2) control_latent_input = torch.cat([unet_latent_input] * 2)
if cfg_injection: # only applying ControlNet to conditional instead of in unconditioned if cfg_injection: # only applying ControlNet to conditional instead of in unconditioned
encoder_hidden_states = torch.cat([conditioning_data.unconditioned_embeddings]) encoder_hidden_states = conditioning_data.text_embeddings
else: else:
encoder_hidden_states = torch.cat([conditioning_data.unconditioned_embeddings, encoder_hidden_states = torch.cat([conditioning_data.unconditioned_embeddings,
conditioning_data.text_embeddings]) conditioning_data.text_embeddings])