当前位置:   article > 正文

手把手教你写COMFYUI 节点,自定义comfyui节点,封装comfyui插件_comfyui插件开发

comfyui插件开发

介绍:

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

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

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

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

最基本的框架:

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

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

  1. class example_class:
  2. @classmethod
  3. def INPUT_TYPES(s):
  4. return {
  5. "required": {
  6. "image": ("IMAGE",),
  7. },
  8. "optional":{
  9. "width":("INT", {"default": 512, "min": 0, "step": 1}),
  10. "type": (
  11. ["ImagePrompt", "PyraCanny", "CPDS", "FaceSwap"],
  12. {"default": "ImagePrompt"},)
  13. }
  14. }
  15. CATEGORY = "example_list/example"
  16. RETURN_TYPES = ("IMAGE",)
  17. FUNCTION = "example_func"
  18. def example_func(self, image,width,type):
  19. return (image,)
  20. NODE_CLASS_MAPPINGS = {
  21. "example": example_class,
  22. }
  23. NODE_DISPLAY_NAME_MAPPINGS = {
  24. "example": "displayname",
  25. }

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

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        就是功能函数,要实现的功能都在这里写就行。

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号