ComfyUI_MIGraphX 插件完全使用指南
1. 插件简介
ComfyUI_MIGraphX 是一个专门为 AMD 显卡用户设计的超级加速插件。这个插件可以让你的 AMD Navi3 和 Navi4 显卡在运行 Stable Diffusion 模型时跑得更快,就像给你的电脑装了个涡轮增压器一样。
插件原地址: https://github.com/pnikolic-amd/ComfyUI_MIGraphX
能干什么: - 让图片生成速度大幅提升(特别是AMD显卡) - 支持多种主流 Stable Diffusion 模型 - 优化内存使用,让你能跑更大的模型
支持的模型: - Stable Diffusion 1.5(最经典的版本) - Stable Diffusion 2.1(改进版) - Stable Diffusion 3.0 medium(最新中等版本) - Stable Diffusion 3.5 medium(更新版本) - SDXL(超高清版本)
2. 如何安装
前置条件(重要!)
在安装插件之前,你需要先安装这些东西,就像盖房子要先打地基一样:
- ROCm(AMD 显卡的驱动支持)
- 带 ROCm 支持的 PyTorch(深度学习框架)
- MIGraphX(AMD 的加速引擎)
安装 MIGraphX:
sudo apt install migraphx
插件安装步骤
-
打开终端(就像打开命令行窗口)
-
进入 ComfyUI 插件文件夹:
cd ComfyUI/custom_nodes
- 下载插件:
git clone https://github.com/pnikolic-amd/ComfyUI_MIGraphX.git
- 进入插件文件夹:
cd ComfyUI_MIGraphX
- 安装依赖:
pip install -r requirements.txt
- 设置性能优化(可选但建议):
export MIGRAPHX_MLIR_USE_SPECIFIC_OPS="attention"
3. 节点详解
3.1 Compile diffusion model on migraphx 节点
这是插件的核心节点,就像一个"模型加速器"。它的作用是把你的 Stable Diffusion 模型"翻译"成 AMD 显卡能更快理解的语言。
在哪里找到这个节点: - 打开 ComfyUI - 右键点击画布 - 选择 "添加节点" → "advanced" → "migraphx" → "Compile diffusion model on migraphx"
3.2 参数详解
| 参数名 (界面显示) | 参数名 (代码中) | 参数值类型 | 建议值 | 通俗解释 (能干嘛的) | 专业解释 | 怎么用/举个例子 |
|---|---|---|---|---|---|---|
| batch_size | batch_size | 整数 | 1 | 一次处理多少张图片,就像洗衣机一次能洗几件衣服 | 批处理大小,决定并行处理的图像数量 | 如果想一次生成4张图,就设置为4 |
| height | height | 整数 | 512 或 1024 | 图片的高度,像素为单位,就像量身高 | 输出图像的垂直像素数 | 512适合快速预览,1024适合高质量输出 |
| width | width | 整数 | 512 或 1024 | 图片的宽度,像素为单位,就像量腰围 | 输出图像的水平像素数 | 必须和height配合,建议用相同值保持正方形 |
| context_len | context_len | 整数 | 77 或 154 | 文字描述的长度限制,就像短信字数限制 | 文本编码器的上下文长度 | 不用T5就用77,用了T5就用154 |
| data_type | data_type | 字符串 | fp16 | 数据精度类型,就像照片的清晰度档位 | 模型权重的数据类型 | fp16速度快内存少,fp32质量高但慢 |
3.3 CompileSD3MIGraphX 节点(专门用于 SD3 模型)
这是专门为 Stable Diffusion 3.0 优化的节点,就像专用工具比万能工具更好用。
| 参数名 (界面显示) | 参数名 (代码中) | 参数值类型 | 建议值 | 通俗解释 (能干嘛的) | 专业解释 | 怎么用/举个例子 |
|---|---|---|---|---|---|---|
| batch_size | batch_size | 整数 | 1 | 一次处理多少张图片,就像打印机一次能打几张 | SD3模型的批处理大小 | 显存够用可以设置2-4提高效率 |
| height | height | 整数 | 1024 | SD3专用高度,就像专门的画布尺寸 | SD3模型推荐的图像高度 | SD3最佳分辨率是1024x1024 |
| width | width | 整数 | 1024 | SD3专用宽度,配合高度使用 | SD3模型推荐的图像宽度 | 保持1024获得最佳质量 |
| context_len | context_len | 整数 | 154 | SD3的文字理解长度,比老版本更长 | SD3的文本编码器上下文长度 | SD3通常需要154来支持更复杂的提示词 |
| data_type | data_type | 字符串 | fp16 | 数据类型,影响速度和质量平衡 | SD3模型的数据精度 | fp16是最佳选择,兼顾速度和质量 |
4. 使用技巧和建议
4.1 连接方式
- 输入端:连接到"Load Checkpoint"节点(模型加载器)
- 输出端:连接到"KSampler"节点(采样器)或其他需要模型的节点
4.2 参数设置技巧
- 尺寸匹配:batch_size、height、width 必须和你的"Empty Latent Image"节点设置一致,就像买衣服要对尺码
- 内存优化:如果显存不够,先把 batch_size 设为1,再尝试降低 height 和 width
- 速度优化:如果追求速度,使用 512x512 分辨率和 fp16 数据类型
4.3 最佳实践
- 第一次使用时,模型编译需要一些时间,就像第一次开新软件需要加载
- 编译完成后,后续使用会非常快
- 建议先用小分辨率测试,确认无误后再用大分辨率
5. 常见问题解答
Q1:为什么我的显卡不支持?
A: 这个插件目前只支持 AMD Navi3 和 Navi4 显卡。如果你的显卡比较老,可能无法使用这个插件。
Q2:安装后找不到节点怎么办?
A: 1. 确认插件安装在正确位置(ComfyUI/custom_nodes/) 2. 重启 ComfyUI 3. 检查依赖是否正确安装 4. 查看 ComfyUI 启动日志是否有错误信息
Q3:编译时间很长正常吗?
A: 第一次编译需要较长时间,这是正常现象。就像第一次做饭需要准备很多材料,但熟练后就很快了。
Q4:context_len 应该设置多少?
A: 如果不使用 T5 编码器就设置 77,使用 T5 编码器就设置 154。
Q5:可以和其他插件一起使用吗?
A: 可以,但建议先单独测试这个插件,确认正常工作后再添加其他插件。
6. 进阶使用建议
6.1 工作流模板
插件提供了预设的工作流模板,你可以在 ComfyUI 中: - 选择 "Workflow" → "Browse Templates" → "ComfyUI_MIGraphX" - 每个模板都针对特定的 Stable Diffusion 模型进行了优化
6.2 性能监控
- 观察显卡使用率,确保充分利用硬件
- 监控显存使用,避免超出限制
- 记录生成时间,对比优化前后的性能提升
6.3 故障排除
如果遇到问题: 1. 检查 ROCm 环境是否正确配置 2. 确认 MIGraphX 版本兼容性 3. 查看插件的 GitHub 页面获取最新更新和解决方案
通过这个插件,AMD 显卡用户也能享受到高效的 AI 图像生成体验。记住,第一次使用可能需要一些耐心,但一旦配置好,你就能体验到显著的性能提升!