Update __init__.py

This commit is contained in:
Extraltodeus
2024-07-20 16:04:04 +02:00
committed by GitHub
parent fd3e0aff9e
commit 172e11cb5c
+2 -2
View File
@@ -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})