Riftai/Rift.1-decoder

[image]

Rift.1-decoder is a VAE decoder made for the Rift model line. It is designed as a drop-in decoder component for compatible Diffusers image pipelines that use AutoencoderKLFlux2. The decoder has been trained as the Rift image reconstruction component.

The exported Diffusers runtime class remains AutoencoderKLFlux2 for loader compatibility. The model metadata identifies the architecture as Rift1Decoder with model type rift1_decoder.

Key Features

1. 1.Diffusers decoder interface using AutoencoderKLFlux2.
2. 2.Rift1Decoder metadata in config.json for clear model identity.
3. 3.32 latent channels for compatible image latent spaces.
4. 4.512px reconstruction training with edge and frequency losses for sharper detail retention.
5. 5.Single-file artifacts included for decoder-focused workflows:
6. 6.diffusion_pytorch_model.safetensors
7. 7.full_encoder_small_decoder.safetensors
8. 8.small_decoder.safetensors
9. 9.Released under the Rift Non-Commercial License v1.0.

Compatible target pipeline family:

• —Diffusers image pipelines using AutoencoderKLFlux2

Comparison

Detail View

[image]

Usage

pip install git+https://github.com/huggingface/diffusers.git transformers accelerate torch

import torch
from diffusers import AutoencoderKLFlux2

vae = AutoencoderKLFlux2.from_pretrained(
    "Rift-ai/Rift.1-decoder",
    torch_dtype=torch.bfloat16,
)

If using a compatible image pipeline, pass this VAE when loading the pipeline:

import torch
from diffusers import AutoencoderKLFlux2

device = "cuda"
dtype = torch.bfloat16

vae = AutoencoderKLFlux2.from_pretrained(
    "Rift-ai/Rift.1-decoder",
    torch_dtype=dtype,
).to(device)

# Pass `vae=vae` into a compatible Diffusers image pipeline.

Artifact Files

Limitations

• —This repository contains a VAE decoder component, not a complete text-to-image model.
• —Visual quality depends on the surrounding diffusion model, scheduler, prompt, latent distribution, and inference settings.
• —The decoder may introduce color shifts, texture smoothing, edge artifacts, or small structural artifacts.
• —Text rendered in generated images may be inaccurate or distorted.
• —Prompt following is handled primarily by the surrounding generation pipeline, not the VAE decoder alone.
• —This model should be evaluated visually and quantitatively before production use.

Out-of-Scope Use

This model and its derivatives may not be used outside the scope of the Rift Non-Commercial License v1.0, including for unlawful, fraudulent, defamatory, abusive, exploitative, privacy-invasive, or otherwise harmful purposes.

Responsible AI Development

Rift.1-decoder should be evaluated as part of a complete image generation or image reconstruction system. A decoder can affect visual fidelity and artifacts, but safety behavior also depends on the text encoder, diffusion transformer, prompt filters, data pipeline, deployment environment, and downstream product policy.

Users are responsible for applying appropriate safeguards, content review, watermarking or provenance notices where required, and compliance with applicable law.

License

This model is licensed under the Rift Non-Commercial License v1.0.

Trademarks & IP

This project may contain trademarks or references to third-party projects, products, or services. Use of Rift, Rift-ai, or associated marks in modified versions of this project must not imply sponsorship, endorsement, approval, or official status unless explicitly authorized. Third-party trademarks, intellectual property, and logos remain subject to their respective owners' policies.