README.md

---
license: other
license_name: nvidia-license
license_link: LICENSE
tags:
  - pixel-diffusion
  - decoder
  - super-resolution
  - non-commercial
  - sceneworks
extra_gated_prompt: >-
  This model is licensed under the NVIDIA License (NSCLv1). It and any
  derivative works may be used for NON-COMMERCIAL (research or evaluation)
  purposes only. The non-commercial restriction flows to decoded output.
---

# PiD — flux2 2kto4k decoder (SceneWorks redistribution)

A format-converted redistribution of NVIDIA's **PiD** (Pixel Diffusion) `flux2`
4-step distillation decoder, packaged for SceneWorks' native (no-PyTorch) PiD
decoder (`mlx-gen-pid` on macOS/MLX, `candle-gen-pid` on Windows·Linux/CUDA).

PiD is an **optional, per-generation replacement for the VAE decoder**: it denoises
directly in pixel space and **decodes + 4× super-resolves in one 4-step pass**. This
`flux2` student serves the **FLUX.2 VAE latent space** — FLUX.2-dev, FLUX.2-klein-9b,
Lens (lens/lens_turbo), and Ideogram 4 (ideogram_4/_turbo). It consumes the packed
128-channel BN-normalized latent (32 raw VAE channels × 2×2 patchify, at H/16).

> **Checkpoint note:** this is the **`_2606`** revision of the 2kto4k flux2 student.
> The un-suffixed 2kto4k flux2 checkpoint has a known color-drift issue and is
> intentionally not used.

## ⚠️ License — non-commercial (research/evaluation) only

This checkpoint is under the **NVIDIA License** (HuggingFace tag `NSCLv1`); see
[`LICENSE`](LICENSE) and [`NOTICE`](NOTICE). Per **§3.3**, the Work and any
derivative works may be used **non-commercially — for research or evaluation
purposes only** (NVIDIA and its affiliates excepted).

**This restriction flows to the output.** Images decoded with PiD are for
research/evaluation use only, distinct from the rest of the SceneWorks pipeline.

## Contents

| File | Description |
|------|-------------|
| `pid_flux2_2kto4k.safetensors` | flux2 `2kto4k` (_2606) 4-step student backbone + sigma-aware LQ adapter (456 tensors, bf16, ~2.7 GB). |
| `LICENSE` | Full NVIDIA License text (verbatim from the original release). |
| `NOTICE` | NVIDIA attribution, provenance, and the exact conversion performed. |

The PiD caption encoder (`gemma-2-2b-it`) and each model's VAE are provisioned
separately by SceneWorks; this repo holds only the PiD student weights.

## Provenance / conversion

Converted from NVIDIA's original
`checkpoints/PiD_res2kto4k_sr4x_official_flux2_distill_4step_2606/model_ema_bf16.pth`
(from [`nvidia/PiD`](https://huggingface.co/nvidia/PiD)) via a lossless
key/format transform (`net.`-prefix strip + drop training-only
`net_ema.*`/`fake_score.*`/`discriminator.*`; dtype preserved). No re-training.

- Project: https://research.nvidia.com/labs/sil/projects/pid/
- Code: https://github.com/nv-tlabs/pid
- Paper: arXiv:2605.23902