Supra2-IMG safetensors

Safetensors build of SupraLabs/Supra2-IMG. The announcement is Massive Release: Supra2-IMG, a tiny 100M text-to-image model on r/StableDiffusion.

The model is SupraLabs' work: a 104.1M-parameter text-to-image DiT, 256×256, trained from scratch for 10 epochs on LucasFang/FLUX-Reason-6M. The text encoder is frozen Flan-T5-Base. The VAE is SD-VAE-FT-MSE. License is Apache-2.0. Architecture and training details are on the original model card.

This repo

The original checkpoint is model_final_ema.pt. This repo hosts that checkpoint converted to FP32 safetensors.

Supra2-IMG_fp32.safetensors is a safetensors file, not a renamed PyTorch checkpoint. It holds 237 FP32 weight tensors plus the stored unconditional text embedding and mask used for classifier-free guidance. Metadata keeps patch=2 and ctx_len=128.

Flan-T5-Base and SD-VAE-FT-MSE are included under their own names. They are the original models, not part of Supra2. inference.py loads the DiT, the text encoder, the tokenizer, and the VAE from this repo.

Files the script does not read were left out: the SentencePiece model, tokenizer_config.json, special_tokens_map.json, and the original repo config.json. The tokenizer vocab is in flan-t5-base.tokenizer.json.

generate.bat and generate.sh take a prompt and an output filename.

Files

File	Role
inference.py	Sampling script
generate.bat, generate.sh	Launchers
Supra2-IMG_fp32.safetensors	DiT weights, FP32
flan-t5-base.safetensors	Text encoder weights
flan-t5-base.config.json	Text encoder config
flan-t5-base.tokenizer.json	Tokenizer
sd-vae-ft-mse.safetensors	VAE weights
sd-vae-ft-mse.config.json	VAE config

Generate

generate.bat "a boat" boat.png

./generate.sh "a boat" boat.png

A bare filename is written next to the script. A full path is used as given.

python inference.py --prompt "a boat" --seed 0 --cfg 3.0 --steps 50 --out boat.png

Recommended settings from the original model card: seed 0, CFG 3.0, steps 50.