ComfyUI-UltraEdit-ZHO

★ 148

图像编辑模型加载掩码支持自动下载

把 UltraEdit（Diffusers）集成到 ComfyUI，支持本地或自动加载模型，基于图像+蒙版进行可控的定向图像编辑与引导参数调节。

💡 在ComfyUI中用UltraEdit对图片进行基于蒙版的定向编辑。

🍴 6 Forks💻 Python🔄 2024-07-11

🔗 GitHub 原文

📦

网盘下载

复制链接后前往夸克网盘下载

https://pan.quark.cn/s/e00a65475347

📦 requirements.txt

git+https://github.com/HaozheZhao/UltraEdit.git@main#subdirectory=diffusers

📄 README

UltraEdit in ComfyUI

Unofficial implementation of UltraEdit（Diffusers） for ComfyUI

https://github.com/ZHO-ZHO-ZHO/ComfyUI-UltraEdit-ZHO/assets/140084057/074ae7a8-3e51-4df0-ae81-b689437c8631

项目介绍 | Info

对 UltraEdit 的非官方 Diffusers 版实现

UltraEdit：基于 SD3 Medium 的图像编辑模型

允许仅通过提示词实现指定内容的图像编辑，且能保持风格一致性

模型同时支持全局编辑和区域蒙版编辑

版本：V1.0 同时支持本地模型加载（适合本地）和自动下载模型（适合云端）

节点说明 | Features

UltraEdit 模型加载 | 🏕️UltraEdit Model

🏕️UltraEdit Model(local)：加载本地模型，需将 BleachNick/SD3_UltraEdit_w_mask 中的所有模型和文件手动下载放入 /ComfyUI/models/ultraedit 中

🏕️UltraEdit Model(auto)：会自动下载模型，请保持网络畅通

UltraEdit 生成 | 🏕️UltraEdit Generation

pipe：接入模型

image：接入图片

mask：非必要项，接入蒙版（需图片格式，可通过 convert mask to image 节点转换）

positive：正向提示词

negative：负向提示词

step：步数，默认 50 步（模型是 512*512）

image_guidance_scale：图像相关度，默认为 1.5

text_guidance_scale：文本引导相关度，默认为 7.5

seed：种子

使用注意 | Attention

输入图像尺寸需是 4 的倍数

如果得到的输出是 512 张图而非单张图像，那需要手动将 UltraEdit.py 文件最后的 output_t = output_t.squeeze(0) 删除，我自己的测试均未出现这个问题，目前还不清楚是啥原因（估计还是 diffusers 问题）

安装 | Install

注意：此项目需要特殊的 diffusers 版本，推荐使用虚拟环境或云（避免冲突），可直接通过 requirements 自动安装依赖，也可手动安装：pip install git+https://github.com/HaozheZhao/UltraEdit.git@main#subdirectory=diffusers

推荐使用管理器 ComfyUI Manager 安装

手动安装：

cd custom_nodes

git clone https://github.com/ZHO-ZHO-ZHO/ComfyUI-UltraEdit-ZHO.git

cd custom_nodes/ComfyUI-UltraEdit-ZHO

pip install -r requirements.txt

重启 ComfyUI