ColoDiffusion: Stable Diffusion with Colossal-AI

Roadmap

Installation

Option #1: install from source

Step 1: Requirements

Step 2:Install Colossal-AI From Our Official Website

Download suggested verision for this training

Download the latest version from pip for latest torch version

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

Invitation to open-source contribution

Comments

BibTeX