AlphaVAE: Unified End-to-End RGBA Image Reconstruction and Generation with Alpha-Aware Representation Learning
Paper • 2507.09308 • Published
Extract foregrounds as transparent PNGs, including soft edges, shadows and partial transparency.
| Download | Model |
|---|---|
vae/ |
RGBA VAE for FLUX.2 |
loras/extract_4b.safetensors |
Foreground extractor for Klein Base 4B |
loras/extract_9b.safetensors |
Foreground extractor for Klein Base 9B |
loras/remove_9b.safetensors |
Object removal for Klein Base 9B |
The demo includes a 4B / 9B extractor selector and an auto-background mode: mark an object, generate its background, then extract it as RGBA. You can also provide a composite and background directly.
hf download trmz/flux2-klein-alpha --local-dir weights
Follow the demo setup. Choose 4B or 9B under Extractor model. Start directly with 4B using EXTRACT_MODEL=4B.
Use prompt Foreground, guidance 4. Both extractors run at six steps.
By Xavier Jara. VAE method: AlphaVAE. Training/inference: ai-toolkit.
Base model
black-forest-labs/FLUX.2-klein-base-4B