stable-diffusion-aws-extension/docs/zh/user-guide/ComfyUI/inference.md

7.3 KiB
Raw Permalink Blame History

在comfyUI上推理

在成功部署解决方案后,您可以打开部署成功的堆栈所提供的ComfyUIDesigner端口页面,在工作流的调试、发布和推理,总结步骤如下:

  • 步骤1: 连接部署了ComfyUI前端的EC2。
  • 步骤2: 在本方案提供的Designer链接页面可进行新工作流的调试安装缺失的节点并上传所需的推理模型并能够成功在本地虚拟机EC2加载并推理成功发布此工作流为模版template
  • 步骤3: 按需重复步骤2以发布多个工作流模版template
  • 步骤4: 在调试完一个或多个工作流模版template一键打包发布当前环境并配合部署新的Amazon SageMaker推理节点。
  • 步骤5: 在ComfyUI推理页面选择已发布模版按需修改推理词、推理模型并推理图片/视频。本步骤将会使用Amazon SageMaker资源。

!!! tip "贴士" 用户可以创建一个或多个包含相近功能或插件组合的工作流模版template。调试完这些模版后将当前的并集环境打包成一个运行环境environment并为其配置推理节点Amazon SageMaker Endpoint。这种方法确保了工作流的稳定性和兼容性。遇到部署问题时可以查看“常见问题解答”章节中ComfyUI相关部分。

步骤1: 连接部署ComfyUI前端的虚拟机EC2

通过连接EC2的方式在本地打开EC2对应的目录结构能更方便的看到本地工作流调试时的日志文件帮助判断。

如仅需看到本地ComfyUI调试的日志也可通过以下步骤实现

  1. 打开部署解决方案同region的EC2控制台选择comfy-on-aws-dev机器,并点击右上角Connect
  2. 在可选的连接方式中,选择标签页EC2 Instance Connect,点击Connect
  3. 稍作等待即可看到新弹出的、EC2链接页面。按需输入代码即可完成各类操作。常用命令如下
tail -f /var/log/cloud-init-output.log 用于实时查看ec2启动Comfy过程初始日志
sudo journalctl -u comfy -f 用于实时查看comfy运行日志
tail -f /root/stable-diffusion-aws-extension/container/*.log 用于查看comfy运行容器所有日志
sudo journalctl -u comfy --no-pager -n 200 用于查看comfy运行日志最后200条
docker images -q | xargs docker rmi -f

步骤2: 调试工作流并发布为模版

打开本方案部署后,提供的Designer链接,即为工作流调试员/主美视角的ComfyUI您可以按单机版ComfyUI的使用方法来进行新的工作流的调试。有关模型的管理、定制节点的管理等可以通过连接部署ComfyUI的虚拟机EC2来进行。

major-designer

主美调试的ComfyUI使用步骤总结如下

  1. 可选拖拽已有工作流的json文件到ComfyUI界面得到渲染出来的工作流。

  2. 调整包括添加和删除工作节点custom node并调整推理参数及使用的模型。

  3. 点击添加提示词队列Queue Prompt,开启基于当前页面工作流的推理任务。

  4. 可选当步骤3遇到弹框提示报错时需要按指示处理。比如若显示某模型缺失请先将该模型下载至EC2对应的文档目录如果显示有定制节点的缺失请点击管理器Manager**安装缺失节点Install Missing Custom Nodes**来安装缺失节点。按错误提示处理完成后重复步骤3进行测试。 ComfyUI Management

  5. 当工作流完成后,看到界面显示出生成结果,即表示该工作流的调试工作已经成功。

  6. 点击右侧导航栏中工作流模版列表模块上方加号,在弹框中填入待发布模版的名字,及需要绑定的运行环境,点击OK

    !!! tip "贴士" 新模版命名长度不能超过20个字符英文与数字结合大小写敏感。同时该命名需要是同一个region内唯一如果与现有模版名字冲突会提示创建错误。另外此处的运行环境如还未发布可以等到步骤4完成后返回修改已发布的模版以绑定正确的运行环境。

template release

  1. 工作流发布过程中ComfyUI前端不可以有更新操作该过程大致持续十几秒。在发布完成后会在前端有弹框提示发布完成。
  2. 发布完成,刷新页面即可在右侧导航栏的工作流模版列表看到新发布的模版。

步骤3: 按需重复步骤2以发布多个模版

为确保工作流的稳定性和兼容性以及最大化资源利用率建议用户在ComfyUI的默认环境中持续调试并发布多个工作流模版template。在调试完成后将当前的并集运行环境发布并绑定到一个Amazon SageMaker推理节点Inference Endpoint。这样发布的环境可以稳定运行多个绑定的模板最大化推理节点的资源利用率。同时需要注意发布环境的大小如果环境过大可能会在后续自动弹性伸缩时导致冷启动时间过长。用户需要在多个模板绑定一个环境的数量上做出权衡。

步骤4: 发布环境并创建推理节点

待步骤3完成后用户可以一键打包发布工作流运行环境并创建推理端点来依据工作流进行云端推理。

  1. 点击右侧导航栏中New Environment.

  2. 在弹框中填入待发布环境的名字。在Endpoint Config区域选择为该环境绑定的Amazon SageMaker Inference Endpoint参数包括机型、是否自动扩展等。完成后点击OK

    !!! Important "注意" 如果是相对复杂的workflow 注意选择异步推理节点类型否则受限于service最长等待30s会出现调用推理超时的情况。另外不建议直接进去SageMaker console直接删除endpoint容易造成数据不一致的隐患。

  3. 发布环境过程会基于现有环境大小等不定时间可能长达10+分钟。等待期间,页面锁住,不可以进行其他操作,以防环境被破坏导致发布失败。

  4. 创建成功后,请刷新页面,在Template List区域选中需要绑定的模版,点击铅笔图标,在弹框中选择新发布环境,点击OK

步骤5: 基于已发布工作流的推理

在ComfyUI的推理页面可以简单通过以下步骤完成基于模版的推理。

  1. 打开ComfyUI的推理页面在右侧导航栏选择一个已发布的模版。如果是调试环境还需要选中右侧导航栏中Prompt on AWS复选框。

  2. ComfyUI页面即会自动渲染该模版的工作流按需调整参数并点击添加提示词队列Queue Prompt,提交推理任务。

  3. 当推理任务完成后,生成结果会自动展示在页面。

    !!! Important "注意" 图生图时 如果涉及到新的上传图片到input文件夹以后 系统会自动同步到当前环境对应的endpoint但可能需要稍等1min后再试。

模型管理

模型上传

新模型需要上传到EC2才能保证在模型调试阶段能够顺利调取。在EC2中上传模型可以通过进入models目录下对应的模型类别子文件夹通过直接拖拽或者wget + 模型下载地址的方式来实现。考虑到网速推荐优先使用wget模型下载方式。