Update __init__.py
This commit is contained in:
+2
-2
@@ -18,10 +18,10 @@ NODE_CLASS_MAPPINGS = {
|
||||
"Support empty uncond": support_empty_uncond_pre_cfg_node,
|
||||
"Shape attention": ShapeAttentionNode,
|
||||
"Post CFG subtract mean": PostCFGsubtractMeanNode,
|
||||
"Post CFG make a dot": PostCFGDotNode,
|
||||
# "Post CFG make a dot": PostCFGDotNode,
|
||||
"Individual channel selector": individual_channel_selection_node,
|
||||
|
||||
}
|
||||
|
||||
for c in [4,8,16,32,64,128]:
|
||||
NODE_CLASS_MAPPINGS[f"Channel selector for {c} channels"] = type("channel_selection_node", (channel_selection_node,), { "CHANNELS_AMOUNT": c})
|
||||
NODE_CLASS_MAPPINGS[f"Channel selector for {c} channels"] = type("channel_selection_node", (channel_selection_node,), { "CHANNELS_AMOUNT": c})
|
||||
|
||||
Reference in New Issue
Block a user