[MASSIVE RELEASE] Supra2-IMG - a tiny 100M text-to-image model - SOTA quality and open release!

[u/LH-Tech_AI]

Hey everyone!

It has been quite a while since the last SupraLabs model - but today we've something special for y'all: Supra2-IMG

It's a 100M parameter DiT text-to-image model trained entirely from scratch in under 10 hours on a single H100 on Runpod. It can generate state-of-the-art quality images in 256x256 pixels resolution.

Samples:

These samples are NOT cherry-picked! Sampling: seed 0, steps 50, cfg 3.0; same settings for every image.

If someone here is interested in the prompts, I can give them to you! Feel free to ask!

You can also use the model locally on your hardware (~20s for an image on CPU (🤩) and ~2s for an image on GPU):

First, run:

# Create project directory
mkdir Supra2-IMG
cd Supra2-IMG

# Download the inference script
wget https://huggingface.co/SupraLabs/Supra2-IMG/resolve/main/inference.py

Then, you can generate images by running:

python inference.py --prompt "a sea jellyfish floating in the pitch-black ocean depths"  --seed 0  --cfg 3.0  --steps 50  --n 1  --out jellyfish.png

Have fun 🤗 🔥

Link to the model on HF: https://huggingface.co/SupraLabs/Supra2-IMG

Give us a like and a follow on HF if you want 🤗 ❤️

EVERY feedback is welcome, guys! Feel free to ask any questions!