Skip to content

Commit 75ab27b

Browse files
committed
UnionInput set item
1 parent 18eaa77 commit 75ab27b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/diffusers/models/controlnets/controlnet_union.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,8 @@ def __iter__(self):
7070
def __getitem__(self, key):
7171
return getattr(self, key)
7272

73+
def __setitem__(self, key, value):
74+
setattr(self, key, value)
7375

7476
@dataclass
7577
class ControlNetUnionInputProMax:
@@ -104,6 +106,8 @@ def __iter__(self):
104106
def __getitem__(self, key):
105107
return getattr(self, key)
106108

109+
def __setitem__(self, key, value):
110+
setattr(self, key, value)
107111

108112
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
109113

0 commit comments

Comments
 (0)