File size: 733 Bytes
8ebda9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# 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
```
|