ComfyUI-Open-Sora

★ 106

图像生成模型集成Linux专用高质量输出

将Open Sora模型集成到ComfyUI，提供Linux下依赖安装与模型加载方案，便捷生成高质量图像。

💡 在ComfyUI上配置并使用Open Sora模型生成高质量图像

🍴 17 Forks💻 Python🔄 2024-07-19

🔗 GitHub 原文

📦

网盘下载

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

https://pan.quark.cn/s/2df45d172dc1

📦 requirements.txt

colossalai
accelerate
diffusers
ftfy
gdown
mmengine
pre-commit
pyav
tensorboard
timm
tqdm
transformers
wandb

📄 README

ComfyUI-Open-Sora

only supports Linux now

pip install packaging ninja
pip install flash-attn --no-build-isolation

git clone https://www.github.com/nvidia/apex
cd apex
sudo python setup.py install --cuda_ext --cpp_ext

pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121

cd ComfyUI/custom_nodes
git clone https://github.com/chaojie/ComfyUI-Open-Sora
cd ComfyUI-Open-Sora
pip install -v .

Models

download OpenSora-v1-HQ-16x512x512.pth & put it to ComfyUI/models/checkpoints

| ———- | —— | ———– | ———- | ————— | ——————————————————————————————— |

| 16×256×256 | 366K | 80k | 8×64 | 117 | :link: |

| 16×256×256 | 20K HQ | 24k | 8×64 | 45 | :link: |

| 16×512×512 | 20K HQ | 20k | 2×64 | 35 | :link: |

pip install -U huggingface_hub
中国用户可以执行这一步 export HF_ENDPOINT=https://hf-mirror.com  windows上在powershell里$env:HF_ENDPOINT = "https://hf-mirror.com"
huggingface-cli download --resume-download DeepFloyd/t5-v1_1-xxl
huggingface-cli download --resume-download stabilityai/sd-vae-ft-ema

Examples

base workflow

https://github.com/chaojie/ComfyUI-Oepn-Sora/blob/main/wf.json

Open-Sora