Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix condtionings logic

Browse Source
This commit is contained in:
Sergey Borisov 2024-07-16 00:31:56 +03:00
parent 499e4d4fde
commit d623bd429b
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/backend/stable_diffusion/diffusion/conditioning_data.py
View File

@ -131,16 +131,18 @@ class TextConditioningData:
# TODO: combine regions with conditionings
if conditioning_mode == "both":
conditionings = [self.uncond_text.embeds, self.cond_text.embeds]
conditionings = [self.uncond_text, self.cond_text]
c_regions = [self.uncond_regions, self.cond_regions]
elif conditioning_mode == "positive":
conditionings = [self.cond_text.embeds]
conditionings = [self.cond_text]
c_regions = [self.cond_regions]
else:
conditionings = [self.uncond_text.embeds]
conditionings = [self.uncond_text]
c_regions = [self.uncond_regions]
encoder_hidden_states, encoder_attention_mask = self._concat_conditionings_for_batch(conditionings)
encoder_hidden_states, encoder_attention_mask = self._concat_conditionings_for_batch(
[c.embeds for c in conditionings]
)
unet_kwargs.encoder_hidden_states = encoder_hidden_states
unet_kwargs.encoder_attention_mask = encoder_attention_mask