全网最全面的comfyui节点安装教程！手把手教你写COMFYUI 节点，自定义comfyui节点，封装comfyui插件

AI绘画小33

已于 2024-09-02 10:54:55 修改

阅读量1.9k

点赞数 18

文章标签： python 开发语言人工智能 stable diffusion AIGC comfyui

于 2024-09-02 10:50:11 首次发布

本文链接：https://blog.csdn.net/2401_84760527/article/details/141814115

版权

前言

介绍：

众所周知，comfyui是一个AI绘图领域的一个十分好用的工具，可以完成很多十分复杂的工作流。工作效率远远高于传统的webui。同时，入门门槛十分低，在bilibili看一个小时教程即可完全入门。

所以很多工作室将comfyui工作流作为stable diffusion开发的主要工具。

本文即来教大家如何使用comfyui写自己的插件。保证通俗易懂，看完就会。

（当然默认你已经安装了comfyui，并且知道怎么使用。并且了解python简单的语法）

本次教程将使用AI绘画工具 Stable Diffusion comfyui进行讲解，如还未安装comfyui的小伙伴可以扫描免费获取哦~

最基本的框架：

首先，在comfyui文件目录下，custom_nodes 文件夹是我们要放置我们的插件的地方。

下面我将演示一下，如何使用代码写一个最简单的插件。首先，新建一个py文件，命名为example.py

class example_class:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "image": ("IMAGE",),
            },
            "optional":{
                "width":("INT", {"default": 512, "min": 0, "step": 1}),
                "type": (
                    ["ImagePrompt", "PyraCanny", "CPDS", "FaceSwap"],
                    {"default": "ImagePrompt"},)


}
        }

    CATEGORY = "example_list/example"

    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "example_func"

    def example_func(self, image,width,type):
        return (image,)



NODE_CLASS_MAPPINGS = {
    "example": example_class,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "example": "displayname",

}

无论写什么节点，都可以先无脑把这套框架搬运过去。下面我将一行行介绍这些代码，当然懂代码的可能看到这里就已经懂了。

input 函数

要写一个节点，首先要声明一个类，class，然后要写的第一个函数就是INPUT_TYPES 这个函数。他要控制我们写的节点中输入是什么。

首先，require里要写必须要连接的节点，在comfyui中如果不连接就会报错，optional是可选的选项，

如果我们节点的输入是类型是上一个节点输出的类型，“image”: (“IMAGE”,),就像这样就可以了。

括号里写输出节点的输出类型就OK了。

如果节点的输入类型是，int数字或者float 数字，那么就像这样。

“width”😦“INT”, {“default”: 512, “min”: 0, “step”: 1}),

default 是默认的数字，min是最小值，max 是最大值，step是comfyui里左右调整时的步长。

如果输入是选择题，就像这样。

            "type": (  
                \["ImagePrompt", "PyraCanny", "CPDS", "FaceSwap"\],  
                {"default": "ImagePrompt"},)

其他参数：

CATEGORY 控制这个节点你在comfyui右键后，存放的位置。

RETURN_TYPES :输出的节点的类型

example_func 就是功能函数，要实现的功能都在这里写就行。

为了帮助大家更好地掌握 ComfyUI，我在去年花了几个月的时间，撰写并录制了一套ComfyUI的基础教程，共六篇。这套教程详细介绍了选择ComfyUI的理由、其优缺点、下载安装方法、模型与插件的安装、工作流节点和底层逻辑详解、遮罩修改重绘/Inpenting模块以及SDXL工作流手把手搭建。