diff --git a/README.md b/README.md index 5230e8a..456fd34 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,43 @@ # ComfyUI_RH_OminiControl -The OminiControl plugin for ComfyUI + +**ComfyUI_RH_OminiControl** 是一个 **ComfyUI** 的插件,基于 **OminiControl**,旨在通过分拆 pipeline 加载优化性能,使该项目能够高效运行于 **NVIDIA RTX 4090** 上。同时,使用 **schnell** 模型生成的 **spatial** 和 **fill**,减少采样步数,从而提高计算效率。 + +**ComfyUI_RH_OminiControl** 提供了一种更加高效且易于集成的方式来使用 **OminiControl** 模型。 + +## 功能概述 / Features + +- **高效计算 / Efficient Computation**: 通过分拆 pipeline 加载优化性能。 +- **schnell 模型支持 / Schnell Model Support**: 使用更少的采样步数以提高效率。 +- **ComfyUI 插件集成 / ComfyUI Plugin Integration**: 与 **ComfyUI** 集成,简化使用流程。 +- **FluxPipeline 支持 / FluxPipeline Support**: 与 **diffusers** 库配合使用,支持 **FluxPipeline** 格式。 + +## 安装要求 / Installation Requirements + +大部分所需的库为 **ComfyUI** 常用库,因此无需额外安装其他依赖。但以下库可能需要安装: + +- **diffusers** 版本:0.31.0(支持 FluxPipeline) + +```bash +pip install diffusers==0.31.0 + +## 项目目录结构 / Project Directory Structure + +模型文件路径位于 {comfyui_dir}/models/flux/,以下是项目的目录结构: +{comfyui_dir}/models/flux/ +├── FLUX.1-schnell/ # diffusers 格式的 Flux 模型 +│ └── 下载地址:https://huggingface.co/black-forest-labs/FLUX.1-schnell +│ +├── ComfyUI_RH_OminiControl/ # ComfyUI_RH_OminiControl 插件的模型目录 +│ ├── depth-anything-small-hf/ # 用于深度识别的模型 +│ │ └── 下载地址:https://huggingface.co/LiheYoung/depth-anything-small-hf/tree/main +│ ├── experimental/ # 实验性模型 +│ │ └── 下载地址:https://huggingface.co/Yuanshi/OminiControl/tree/main/experimental +│ └── omini/ # OminiControl 主模型 +│ └── 下载地址:https://huggingface.co/Yuanshi/OminiControl/tree/main/omini + + ![image](https://github.com/user-attachments/assets/c1535de8-5c8d-40e6-aa7e-aac935eb73e9) + +![image](https://github.com/user-attachments/assets/e1200a4d-f483-45de-bec7-0d5f82245dbc) + +![image](https://github.com/user-attachments/assets/4fc66b28-ea82-4507-98b8-2e4320beaf52)