Error when prompt sizes are too different from each other

[antoinedelplace]

Only happens when prompt sizes are too different from each other.

One prompt is : "test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test test "

The other prompt is : "test"

attention-couple-ComfyUI/attention_couple.py

Line 120 in 818b4a2

context_cond = torch.cat([cond for cond in self.negative_positive_conds[1]], dim=0)

 File "/home/ubuntu/doc/ComfyUI/custom_nodes/attention-couple-ComfyUI/attention_couple.py", line 120, in patch
    context_cond = torch.cat([cond for cond in self.negative_positive_conds[1]], dim=0)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: Sizes of tensors must match except in dimension 0. Expected size 154 but got size 77 for tensor number 1 in the list.