diff --git a/__init__.py b/__init__.py index 742b140..1c24038 100644 --- a/__init__.py +++ b/__init__.py @@ -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}) \ No newline at end of file + NODE_CLASS_MAPPINGS[f"Channel selector for {c} channels"] = type("channel_selection_node", (channel_selection_node,), { "CHANNELS_AMOUNT": c})