ColoDiffusion: Stable Diffusion with Colossal-AI

Roadmap

Installation

Option #1: install from source

Step 1: Requirements

Step 2: install lightning

From Source

From pip

Step 3:Install Colossal-AI From Our Official Website

From pip

From source

Step 3:Accelerate with flash attention by xformers(Optional)

Option #2: Use Docker

Download the model checkpoint from pretrained

stable-diffusion-v2-base(Recommand)

stable-diffusion-v1-4

stable-diffusion-v1-5 from runway

Dataset

Training

Training config

Finetune Example

Training on Teyvat Datasets

Inference

Comments

BibTeX