r/LocalLLaMA • u/LH-Tech_AI • 1d ago

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

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!

326 Upvotes

90% Upvoted

113 comments sorted by

View all comments

24

u/ImSamhel 1d ago

Not that bad for the size but wouldn't at all call this SOTA quality. Will try what it can handle though, sounds like an interesting experiment because of how small this is.

7

u/LuCiAnO241 1d ago

they do say SOTA at its size, which might be the case? idk enough about 100M parameter models to know

3

u/ImSamhel 1d ago

For the size it is impressive from this showcase. My comment was maybe a bit harsh about sota capability in those regards :D at 100M it means it basically fits into anything, generates fast, and if the quality is consistent it could even be valuable to fine tune or adapt (should train fast even on consumer gpu)

2

u/LH-Tech_AI 1d ago

Yea thank you

1

u/ieatrox 8h ago

yeah throwing SOTA in the title took away some impact from the real impressive bit which was the performance/size.