AWS Architecture Blog
·
如何在Amazon EKS上使用ComfyUI构建自定义节点工作流
内容提要
ComfyUI是一个开源的节点工作流解决方案，用户可以通过自定义节点扩展功能。本文介绍了在ComfyUI-on-EKS中集成自定义节点的步骤，包括代码准备、环境配置、模型构建、Docker镜像构建和模型上传，帮助用户构建工作流。
关键要点
-
ComfyUI是一个开源的节点工作流解决方案，用户可以通过自定义节点扩展功能。
-
本文介绍了在ComfyUI-on-EKS中集成自定义节点的步骤，包括代码准备、环境配置、模型构建、Docker镜像构建和模型上传。
-
集成自定义节点需要准备自定义节点代码和环境，以及所需模型。
-
自定义节点代码放置在$HOME/ComfyUI/custom_nodes中，环境通过运行pip install -r安装依赖。
-
模型存放在s3://comfyui-models-{account_id}-{region}目录下，触发Lambda函数同步到本地实例存储。
-
构建Docker镜像有两种方式：从GitHub构建和本地构建。
-
从GitHub构建时，需要在Dockerfile中下载每个自定义节点的代码并设置环境和依赖。
-
本地构建时，需要将整个ComfyUI目录复制到Dockerfile中，前提是已有工作正常的ComfyUI环境。
-
上传模型到s3://comfyui-models-{account_id}-{region}目录，GPU节点会自动同步。
-
建议在构建Docker镜像后进行本地测试，以确保其正确运行。
-
使用首选方法对在线K8S pods进行滚动更新，并测试服务。
-
自定义节点使创作者能够充分发挥ComfyUI的潜力，集成多种功能到工作流中。
延伸解读
自定义节点的灵活性与挑战
ComfyUI的自定义节点功能为用户提供了极大的灵活性，允许他们根据需求扩展工作流。然而，集成自定义节点时，用户需注意依赖冲突和环境配置问题，这可能导致开发过程中的复杂性增加。建议在构建Docker镜像前，仔细检查所有自定义节点的依赖关系，以避免运行时错误。
Docker镜像构建方式的选择
在构建Docker镜像时，用户可以选择从GitHub构建或本地构建。虽然从GitHub构建可以更好地控制环境和依赖，但在节点数量较多时，管理和安装过程可能变得繁琐。相对而言，本地构建更为简便，但可能会忽视某些依赖问题，建议用户在构建后进行充分测试。
模型上传与同步注意事项
上传模型至指定的S3目录后，GPU节点会自动同步，但如果模型数量较多，可能需要等待较长时间。用户应定期检查同步进度，确保模型成功上传并可用。此外，使用AWS SSM命令可以帮助用户实时监控同步状态，避免因模型未同步而导致的工作流中断。
延伸问答
ComfyUI是什么？
ComfyUI是一个开源的节点工作流解决方案，用户可以通过自定义节点扩展其功能。
如何在EKS上集成自定义节点？
在EKS上集成自定义节点需要准备自定义节点代码、环境和所需模型，并通过Dockerfile构建镜像。
构建Docker镜像有哪些方法？
构建Docker镜像有两种方法：从GitHub构建和本地构建，分别涉及下载代码和复制本地目录。
如何上传模型到S3？
将模型上传到s3://comfyui-models-{account_id}-{region}目录，GPU节点会自动同步这些模型。
在构建Docker镜像后需要做什么？
建议在构建Docker镜像后进行本地测试，以确保其正确运行。
自定义节点的作用是什么？
自定义节点使创作者能够充分发挥ComfyUI的潜力，集成多种功能到工作流中。
