diff --git a/README.md b/README.md index c6e70fb..bd0a07b 100644 --- a/README.md +++ b/README.md @@ -4,13 +4,15 @@ A ComfyUI custom node that simply integrates the [OOTDiffusion](https://github.c 一个简单接入 OOTDiffusion 的 ComfyUI 节点。 +👇 拖动下面的图片到 ComfyUI 前端即可运行 👇 Drag into ComfyUI frontend + ![](./assets/graph.png) ## Instruction 指南 根据 https://git-lfs.com 安装 git lfs: -Linux: +Ubuntu / Debian: ```txt sudo apt install git-lfs @@ -22,13 +24,13 @@ git lfs 初始化: git lfs install ``` -拉取 Huggingface 库至 ComfyUI 根目录下的 `models/OOTDiffusion` 目录: +拉取 huggingface 🤗 库至 ComfyUI 根目录下的 `models/OOTDiffusion` 目录: ```txt git clone https://huggingface.co/levihsu/OOTDiffusion models/OOTDiffusion ``` -拉取 Huggingface 时大约会下载 8 个模型,假如断开连接,可以使用下面命令恢复下载: +拉取 huggingface 时大约会下载 8 个模型,假如断开连接,可以使用下面命令恢复下载: ```txt cd models/OOTDiffusion @@ -41,10 +43,9 @@ git checkout main ```txt conda create -n ootd conda activate ootd -conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia -# 修复潜在的一些问题 -conda install cudatoolkit-dev -c conda-forge +# 选择安装 11.8 / 12.1 cuda toolkit +conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia # 安装此项目的依赖 pip install -r custom_nodes/ComfyUI-OOTDiffusion/requirements.txt @@ -52,21 +53,37 @@ pip install -r custom_nodes/ComfyUI-OOTDiffusion/requirements.txt 启动 ComfyUI 即可。 -## 节点 +## FAQ 常见错误 -LoadOOTDPipeline: 加载 OOTD Pipeline +> OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root. +> +> 解决方法:`conda install -c conda-forge cudatoolkit-dev`。 +> 参照:https://github.com/conda/conda/issues/7757 -OOTDGenerate: 生成图像 +> subprocess.CalledProcessError: Command '['where', 'cl']' returned non-zero exit status 1. +> +> 解决办法:仅在 Windows 下出现,可能需要配置一下 MSVC 编译器。 -## 示例图片 +## Node 节点 + +Load OOTDiffusion: 加载 OOTDiffusion Pipeline + +OOTDiffusion Generate: 生成图像 + +## Example image 示例图片 [衣服 1](./assets/cloth_1.jpg) [模特 1](./assets/model_1.png) -## 注意事项 +## Detail 细节 目前此项目只是对 OOTDiffusion 的功能做了个简单的迁移。 OOTDiffusion 本体依赖于 `diffusers==0.24.0` 实现,所以假如有其他节点的依赖冲突是没办法解决的(本就不该依赖 diffusers)。 +靠 vendor 也能解决,所以也不是大问题。 + +不使用 huggingface_hub 是因为 OOTD 提供的仓库并不是一个单纯的 diffusion model structure, +里面还包含了独立的 openpose 和 humanparsing 模型文件。 +目前只有 openai/clip-vit-large-patch14 是使用 huggingface_hub 下载的。 在 `Ubuntu 22.02` / `Python 3.10.x` 下可以正常运行。Windows 没有测试过。 diff --git a/__init__.py b/__init__.py index 73e03fb..4419725 100644 --- a/__init__.py +++ b/__init__.py @@ -86,3 +86,8 @@ NODE_CLASS_MAPPINGS = { "LoadOOTDPipeline": LoadOOTDPipeline, "OOTDGenerate": OOTDGenerate, } + +NODE_DISPLAY_NAME_MAPPINGS = { + "LoadOOTDPipeline": "Load OOTDiffusion", + "OOTDGenerate": "OOTDiffusion Generate", +} diff --git a/assets/graph.png b/assets/graph.png index 33514f4..e0ef30f 100644 Binary files a/assets/graph.png and b/assets/graph.png differ