Skip to content

ComfyUI custom node that simply integrates the OOTDiffusion.

License

Notifications You must be signed in to change notification settings

martyyz2112/ComfyUI-OOTDiffusion

 
 

Repository files navigation

ComfyUI OOTDiffusion

A ComfyUI custom node that simply integrates the OOTDiffusion functionality.

一个简单接入 OOTDiffusion 的 ComfyUI 节点。

Example workflow: workflow.json

Instruction 指南

Create environment and download dependencies:

创建环境并下载依赖:

conda create -n ootd python=3.10
conda activate ootd

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

# Install nvcc compiler for torch cpp extensions
conda install cuda-nvcc -c nvidia

# Clone to custom_nodes
git clone https://github.com/AuroBit/ComfyUI-OOTDiffusion.git custom_nodes/ComfyUI-OOTDiffusion

# Install dependencies
pip install -r custom_nodes/ComfyUI-OOTDiffusion/requirements.txt

Simply start ComfyUI and drag the example workflow.json.

启动 ComfyUI 并拖入示例 workflow.json 即可。

Windows 指南

Windows 除了需要参照上面的配置教程,还需要额外配置 MSVC:

首先是 nvcc 安装(确保 VIRTUAL_ENV/bin/nvcc 存在):

conda install cuda-nvcc -c nvidia

根据自己系统选择 Windows 10 SDK / Windows 11 SDK.

注意:不要从 Installer 点击启动 terminal

进入 C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\VC\Auxiliary\Build 目录, 右击进入 terminal,注意要用 CMD 不要 powershell。

启动 native x64 target x64 的编译环境变量(兼容环境下 nvcc 编译会报错):

vcvars64.bat

编译完成,成功启动。

diffusers 版本

main 分支锁定 diffusers==0.24

diffusers-0.26 分支锁定 diffusers==0.26.x

要切换分支,请使用下面命令:

git switch diffusers-0.26

并重新安装依赖:

pip install --force-reinstall -r custom_nodes/ComfyUI-OOTDiffusion/requirements.txt

FAQ 常见错误

fatal error: cuda_runtime.h: No such file or directory compilation terminated.
ninja: build stopped: subcommand failed.

解决办法:conda install cuda-toolkit=12.1 -c nvidia 并覆写 CUDA_HOME CUDA_PATH 环境变量

subprocess.CalledProcessError: Command '['where', 'cl']' returned non-zero exit status 1.

解决办法:仅在 Windows 下出现,根据 Windows 配置教程

Node 节点

Load OOTDiffusion Local: 从本地目录加载 OOTDiffusion Pipeline

Load OOTDiffusion from Hub: 从 huggingface 自动下载并加载 OOTDiffusion Pipeline

OOTDiffusion Generate: 生成图像

参数:

cfg: 输出图像和输入衣服的贴合程度

Example image 示例图片

Half body 半身: 模特 衣服

Full body 全身: 模特 裤子

Full body 裙子: 模特 裙子

更新日志 Release Note

2024-03-14:

添加 diffusers-0.26 分支

2024-03-10:

添加 humanparsing onnx 支持

2024-03-04:

添加 Full body 模型

2024-03-01:

添加 Windows 安装教程

2024-02-25:

移除 git lfs 下载教程,提供 Load OOTDiffusion from Hub 节点

About

ComfyUI custom node that simply integrates the OOTDiffusion.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 91.3%
  • Cuda 5.2%
  • C++ 2.9%
  • Shell 0.5%
  • Dockerfile 0.1%
  • C 0.0%