Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
stable-diffusion-diffusers
Instructions to use nota-ai/bk-sdm-tiny-2m with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use nota-ai/bk-sdm-tiny-2m with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("nota-ai/bk-sdm-tiny-2m", torch_dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Commit ·
aad3e0e
1
Parent(s): dfed9a6
Update README.md
Browse files
README.md
CHANGED
|
@@ -124,7 +124,7 @@ Follow [the usage guidelines of Stable Diffusion v1](https://huggingface.co/Comp
|
|
| 124 |
# Citation
|
| 125 |
```bibtex
|
| 126 |
@article{kim2023architectural,
|
| 127 |
-
title={
|
| 128 |
author={Kim, Bo-Kyeong and Song, Hyoung-Kyu and Castells, Thibault and Choi, Shinkook},
|
| 129 |
journal={arXiv preprint arXiv:2305.15798},
|
| 130 |
year={2023},
|
|
@@ -132,7 +132,7 @@ Follow [the usage guidelines of Stable Diffusion v1](https://huggingface.co/Comp
|
|
| 132 |
}
|
| 133 |
```
|
| 134 |
```bibtex
|
| 135 |
-
@article{
|
| 136 |
title={BK-SDM: Architecturally Compressed Stable Diffusion for Efficient Text-to-Image Generation},
|
| 137 |
author={Kim, Bo-Kyeong and Song, Hyoung-Kyu and Castells, Thibault and Choi, Shinkook},
|
| 138 |
journal={ICML Workshop on Efficient Systems for Foundation Models (ES-FoMo)},
|
|
|
|
| 124 |
# Citation
|
| 125 |
```bibtex
|
| 126 |
@article{kim2023architectural,
|
| 127 |
+
title={BK-SDM: A Lightweight, Fast, and Cheap Version of Stable Diffusion},
|
| 128 |
author={Kim, Bo-Kyeong and Song, Hyoung-Kyu and Castells, Thibault and Choi, Shinkook},
|
| 129 |
journal={arXiv preprint arXiv:2305.15798},
|
| 130 |
year={2023},
|
|
|
|
| 132 |
}
|
| 133 |
```
|
| 134 |
```bibtex
|
| 135 |
+
@article{kim2023bksdm,
|
| 136 |
title={BK-SDM: Architecturally Compressed Stable Diffusion for Efficient Text-to-Image Generation},
|
| 137 |
author={Kim, Bo-Kyeong and Song, Hyoung-Kyu and Castells, Thibault and Choi, Shinkook},
|
| 138 |
journal={ICML Workshop on Efficient Systems for Foundation Models (ES-FoMo)},
|