nodes/eastmoe-ComfyUI-CMP-Extention
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改节点注册位置

Browse Source
This commit is contained in:
eastmoe
2026-06-29 17:02:39 +08:00
parent 832e310c98
commit 76d86963ba
3 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+6 -6
View File
@@ -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
README_zh.md
+6 -6
View File
@@ -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"
```
## 许可证
__init__.py
+3 -3
View File
@@ -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,