# 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](https://huggingface.co/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 | |
``` | |