SSD-1B-colab

🐣 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
	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

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)

https://github.com/camenduru/SSD-1B-colab/assets/54370274/499175b9-0742-4d94-8961-e2ca00bd6684

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)

https://github.com/camenduru/SSD-1B-colab/assets/54370274/924ca0db-0e00-4a5c-8194-759f23bbe294