From cf697d1b4d612ed35afd4a87f36a759d323c7702 Mon Sep 17 00:00:00 2001 From: nullquant Date: Fri, 3 May 2024 16:13:25 +0300 Subject: [PATCH] pooled_output error --- brushnet_nodes.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/brushnet_nodes.py b/brushnet_nodes.py index 14cb0c6..e9c0032 100644 --- a/brushnet_nodes.py +++ b/brushnet_nodes.py @@ -251,8 +251,21 @@ class BrushNet: prompt_embeds = positive[0][0][0].to(brushnet['brushnet'].device) negative_prompt_embeds = negative[0][0][0].to(brushnet['brushnet'].device) - add_text_embeds = positive[0][1]['pooled_output'].to(brushnet['brushnet'].device) - negative_pooled_prompt_embeds = negative[0][1]['pooled_output'].to(brushnet['brushnet'].device) + if len(positive[0]) > 1 and 'pooled_output' in positive[0][1]: + add_text_embeds = positive[0][1]['pooled_output'].to(brushnet['brushnet'].device) + else: + print('BrushNet: positive conditioning has not pooled_output') + if is_SDXL: + print('BrushNet will not produce correct results') + add_text_embeds = torch.empty([2, 1280], device=brushnet['brushnet'].device) + + if len(negative[0]) > 1 and 'pooled_output' in negative[0][1]: + negative_pooled_prompt_embeds = negative[0][1]['pooled_output'].to(brushnet['brushnet'].device) + else: + print('BrushNet: negative conditioning has not pooled_output') + if is_SDXL: + print('BrushNet will not produce correct results') + negative_pooled_prompt_embeds = torch.empty([1, 1280], device=brushnet['brushnet'].device) add_time_ids = torch.FloatTensor([[height, width, 0., 0., height, width]]).to(brushnet['brushnet'].device) negative_add_time_ids = add_time_ids