Make Attention Couple node accept a single region from Attention Couple or a regions list from Attention Couple Regions node

resolves #3
2024-06-29 04:20:46 +03:00
parent 74567539de
commit 67df043bba
1 changed files with 3 additions and 0 deletions
@@ -158,6 +158,9 @@ class AttentionCouple:

        new_model = model.clone()

+        if not isinstance(regions, list):
+            regions = [regions]
+
        num_conds = len(regions) + 1

        mask = [base_mask] + [