UE封面图正式版

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 同时支持本地模型加载（适合本地）和自动下载模型（适合云端）

screenshot-20240710-033335

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：种子

screenshot-20240710-183111

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

注意：此项目需要特殊的 diffusers 版本，推荐使用虚拟环境或云（避免冲突），可直接通过 requirements 自动安装依赖，也可手动安装：pip install git+https://github.com/HaozheZhao/UltraEdit.git@main#subdirectory=diffusers
推荐使用管理器 ComfyUI Manager 安装
手动安装：
1. cd custom_nodes
2. git clone https://github.com/ZHO-ZHO-ZHO/ComfyUI-UltraEdit-ZHO.git
3. cd custom_nodes/ComfyUI-UltraEdit-ZHO
4. pip install -r requirements.txt
5. 重启 ComfyUI