/SSD-1B-colab

Primary LanguageJupyter Notebook

🐣 Please follow me for new updates https://twitter.com/camenduru
🔥 Please join our discord server https://discord.gg/k5BwmmvJJU
🥳 Please join my patreon community https://patreon.com/camenduru

🦒 Colab

Colab Info
Open In Colab SSD_1B_colab

Main Repo

https://github.com/segmind/distill-sd
https://github.com/Nota-NetsPresso/BK-SDM

Paper

https://arxiv.org/abs/2305.15798
https://openreview.net/forum?id=bOVydU0XKC

Model

https://huggingface.co/segmind/SSD-1B

Dataset

https://huggingface.co/datasets/zzliang/GRIT
https://huggingface.co/datasets/wanng/midjourney-v5-202304-clean

Output

Screenshot 2023-10-24 153135

segmind/SSD-1B vs stabilityai/stable-diffusion-xl-base-1.0

Test 1

1.29 it/s 13.8GB VRAM to 2.07 it/s 8.3GB VRAM (60% Speed 🚀 -39% VRAM Usage With T4)

2023-10-24.16-02-50.mp4

Test 2 Without CPU Offload

1.29 it/s 13.8GB VRAM to 2.07 it/s 11.4GB VRAM (60% Speed 🚀 -17% VRAM Usage With T4)

2023-10-24.14-58-24.mp4