diff --git a/README.md b/README.md index 13c27d7..77ff690 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,7 @@ git clone https://github.com/eastmoe/ComfyUI-CMP-Extention ## Usage -1. In ComfyUI, locate the **CmpExt3ControlPanelNode** within the **CmpExt3** category in the node menu. +1. In ComfyUI, locate the **CmpExt3ControlPanelNode** within the **eastmoe > Comfy-CMP** category in the node menu. 2. Insert this node **after the model loader** and **before the sampler**. 3. Check or uncheck the operator types you wish to optimize as needed. 4. Connect the nodes: @@ -34,8 +34,8 @@ git clone https://github.com/eastmoe/ComfyUI-CMP-Extention ### Core Operators - **Ops: Linear**: Linear layer optimization (Enabled by default) -- **Ops: Conv2d**: Convolutional layer optimization (Enabled by default) -- **Ops: ConvTranspose2d (VAE Upscale)**: Transposed convolution optimization, used for VAE upscaling (Enabled by default) +- **Ops: Conv2d**: Convolutional layer optimization (Disabled by default) +- **Ops: ConvTranspose2d (VAE Upscale)**: Transposed convolution optimization, used for VAE upscaling (Disabled by default) - **Ops: BMM (Attention)**: Batch matrix multiplication optimization in attention mechanisms (Enabled by default) ### Normalization Layers @@ -66,8 +66,8 @@ class CmpExt3ControlPanelNode: "required": { # --- Core --- "enable_linear": ("BOOLEAN", {"default": True, "label": "Ops: Linear"}), - "enable_conv2d": ("BOOLEAN", {"default": True, "label": "Ops: Conv2d"}), - "enable_conv_transpose": ("BOOLEAN", {"default": True, "label": "Ops: ConvTranspose2d (VAE Upscale)"}), + "enable_conv2d": ("BOOLEAN", {"default": False, "label": "Ops: Conv2d"}), + "enable_conv_transpose": ("BOOLEAN", {"default": False, "label": "Ops: ConvTranspose2d (VAE Upscale)"}), "enable_bmm": ("BOOLEAN", {"default": True, "label": "Ops: BMM (Attention)"}), # --- Norms --- @@ -92,7 +92,7 @@ class CmpExt3ControlPanelNode: RETURN_TYPES = ("MODEL", "CLIP", "VAE") RETURN_NAMES = ("model", "clip", "vae") FUNCTION = "execute_patch" - CATEGORY = "CmpExt3" + CATEGORY = "eastmoe/Comfy-CMP" ``` ## License diff --git a/README_zh.md b/README_zh.md index a544cb9..1215002 100644 --- a/README_zh.md +++ b/README_zh.md @@ -17,7 +17,7 @@ git clone https://github.com/eastmoe/ComfyUI-CMP-Extention ## 使用方法 -1. 在 ComfyUI 中,您可以在节点菜单的 **CmpExt3** 分类下找到 **CmpExt3ControlPanelNode** 节点。 +1. 在 ComfyUI 中,您可以在节点菜单的 **eastmoe > Comfy-CMP** 分类下找到 **CmpExt3ControlPanelNode** 节点。 2. 将该节点插入到 **加载模型之后** 与 **采样器之前** 的位置。 3. 根据需要勾选/取消勾选要优化的算子类型。 4. 连接节点: @@ -29,8 +29,8 @@ git clone https://github.com/eastmoe/ComfyUI-CMP-Extention ### 核心算子 - **Ops: Linear**: 线性层优化 (默认启用) -- **Ops: Conv2d**: 卷积层优化 (默认启用) -- **Ops: ConvTranspose2d (VAE Upscale)**: 转置卷积优化,用于VAE上采样 (默认启用) +- **Ops: Conv2d**: 卷积层优化 (默认禁用) +- **Ops: ConvTranspose2d (VAE Upscale)**: 转置卷积优化,用于VAE上采样 (默认禁用) - **Ops: BMM (Attention)**: 注意力机制中的批矩阵乘法优化 (默认启用) ### 归一化层 @@ -61,8 +61,8 @@ class CmpExt3ControlPanelNode: "required": { # --- Core --- "enable_linear": ("BOOLEAN", {"default": True, "label": "Ops: Linear"}), - "enable_conv2d": ("BOOLEAN", {"default": True, "label": "Ops: Conv2d"}), - "enable_conv_transpose": ("BOOLEAN", {"default": True, "label": "Ops: ConvTranspose2d (VAE Upscale)"}), + "enable_conv2d": ("BOOLEAN", {"default": False, "label": "Ops: Conv2d"}), + "enable_conv_transpose": ("BOOLEAN", {"default": False, "label": "Ops: ConvTranspose2d (VAE Upscale)"}), "enable_bmm": ("BOOLEAN", {"default": True, "label": "Ops: BMM (Attention)"}), # --- Norms --- @@ -87,7 +87,7 @@ class CmpExt3ControlPanelNode: RETURN_TYPES = ("MODEL", "CLIP", "VAE") RETURN_NAMES = ("model", "clip", "vae") FUNCTION = "execute_patch" - CATEGORY = "CmpExt3" + CATEGORY = "eastmoe/Comfy-CMP" ``` ## 许可证 diff --git a/__init__.py b/__init__.py index df15b23..e680122 100644 --- a/__init__.py +++ b/__init__.py @@ -294,8 +294,8 @@ class CmpExt3ControlPanelNode: "required": { # --- Core --- "enable_linear": ("BOOLEAN", {"default": True, "label": "Ops: Linear"}), - "enable_conv2d": ("BOOLEAN", {"default": True, "label": "Ops: Conv2d"}), - "enable_conv_transpose": ("BOOLEAN", {"default": True, "label": "Ops: ConvTranspose2d (VAE Upscale)"}), + "enable_conv2d": ("BOOLEAN", {"default": False, "label": "Ops: Conv2d"}), + "enable_conv_transpose": ("BOOLEAN", {"default": False, "label": "Ops: ConvTranspose2d (VAE Upscale)"}), "enable_bmm": ("BOOLEAN", {"default": True, "label": "Ops: BMM (Attention)"}), # --- Norms --- @@ -321,7 +321,7 @@ class CmpExt3ControlPanelNode: RETURN_TYPES = ("MODEL", "CLIP", "VAE") RETURN_NAMES = ("model", "clip", "vae") FUNCTION = "execute_patch" - CATEGORY = "CmpExt3" + CATEGORY = "eastmoe/Comfy-CMP" def execute_patch(self, enable_linear, enable_conv2d, enable_conv_transpose, enable_bmm,