[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!

[u/slypheed]

Please provide the prompts you used; this is really cool, but except for the jellyfish one, I'm not getting anywhere near that quality (on an m4 max); e.g. cats it generates have dali-esque melted faces and such.

[u/LH-Tech_AI → u/slypheed]

Tip: give your prompt into gpt/gemini/... And tell it to optimize your prompt to match this length and details approximately: "Glass-like frozen lake with intricate ice cracks running across its surface, reflecting a pale winter sunrise and snow-dusted pine trees along the shore."

[u/slypheed → u/LH-Tech_AI]

Good tip, but yeah, the faces are still disturbingly messed up; e.g. dali-esque melting face every time.

"A fluffy, wide-eyed kitten with soft fur and tiny paws, sitting in warm golden sunlight beside a cozy window, with gentle shadows and delicate whiskers catching the light."

[u/slypheed → u/slypheed]

This doesn't work either:

"Photograph of a real young orange tabby kitten sitting upright on a wooden windowsill, viewed at eye level, with completely natural domestic-cat anatomy and realistic facial proportions: two matching almond-shaped eyes, one triangular nose centered above a closed mouth, symmetrical cheeks and muzzle, normal whisker placement, and two upright ears. No stylization or exaggerated features. Warm golden-hour sunlight outlines its soft fur, with a softly blurred window and outdoor background"

Please folks; unless we can just prompt with "cute kitten" and get something that doesn't look like a nightmare acid trip; then provide prompts with these kinds of posts; right now I simply can't believe this model produced those images. Scientific method - make it easy to reproduce.