fengshen/examples/disco_project/README.md

Chinese Warp For Disco Diffusion

• This is a chinese version disco diffusion. We train a Chinese CLIP IDEA-CCNL/Taiyi-CLIP-Roberta-large-326M-Chinese and utilize it to guide the diffusion process.
• This code is modified from https://github.com/alembics/disco-diffusion
• streamlit demo is supported.
• the checkpoint has been upload to hugging face.

Usage

• Install the lack package directly

Run Directly

python disco.py --prompt 夕阳西下 --model_path IDEA-CCNL/Taiyi-Diffusion-532M-Nature # or IDEA-CCNL/Taiyi-Diffusion-532M-Cyberpunk

Streamlit Setup

streamlit run st_disco.py
# --server.port=xxxx --server.address=xxxx