Re-host: PiD flux2 2kto4k (_2606) decoder (NVIDIA License / non-commercial) — converted from nvidia/PiD (sc-7852)
ac8592d verified | PiD — Pixel Diffusion Decoder (flux2 2kto4k student) | |
| ===================================================== | |
| This repository redistributes a converted copy of a model checkpoint | |
| originally produced and released by NVIDIA Corporation. | |
| Original work | |
| ------------- | |
| Name: PiD (Pixel Diffusion) — PixelDiT distillation decoders | |
| Author: NVIDIA Corporation and its affiliates (NVIDIA Toronto AI Lab) | |
| Source: https://huggingface.co/nvidia/PiD | |
| Project: https://research.nvidia.com/labs/sil/projects/pid/ | |
| Code: https://github.com/nv-tlabs/pid | |
| Paper: arXiv:2605.23902 | |
| Original checkpoint: | |
| checkpoints/PiD_res2kto4k_sr4x_official_flux2_distill_4step_2606/model_ema_bf16.pth | |
| (the "_2606" revision — the un-suffixed 2kto4k flux2 checkpoint has a | |
| known color-drift issue and is intentionally NOT used.) | |
| Latent space | |
| ------------ | |
| This is the `flux2` PiD student — the FLUX.2 VAE latent space. It consumes the | |
| packed 128-channel BN-normalized latent (32 raw VAE channels × 2×2 patchify, at | |
| H/16). In SceneWorks it serves every model in that latent space: FLUX.2-dev, | |
| FLUX.2-klein-9b, Lens (lens/lens_turbo), and Ideogram 4 (ideogram_4/_turbo). | |
| What was changed | |
| ---------------- | |
| The original PyTorch checkpoint (`model_ema_bf16.pth`) was converted to | |
| safetensors for SceneWorks' native MLX/candle PiD decoder (`mlx-gen-pid`). The | |
| conversion is a lossless key/format transform only: training-only tensors | |
| (`net_ema.*`, `fake_score.*`, `discriminator.*`) are dropped and the `net.` | |
| prefix is stripped, per the reference inference loader | |
| (`pid_distill_model.py::PidDistillModel.load_state_dict`). Tensor values and | |
| dtype (bfloat16) are unchanged. No re-training or fine-tuning was performed. | |
| License | |
| ------- | |
| This work and the original are licensed under the NVIDIA License (the license | |
| HuggingFace tags as "NSCLv1"). The full license text is in the accompanying | |
| LICENSE file and applies to this redistribution and to any derivative works. | |
| USE LIMITATION (NVIDIA License §3.3): The Work and any derivative works thereof | |
| may only be used, or be intended for use, NON-COMMERCIALLY — i.e. for RESEARCH | |
| OR EVALUATION PURPOSES ONLY. (NVIDIA Corporation and its affiliates may use the | |
| Work commercially.) | |
| This non-commercial restriction FLOWS TO THE OUTPUT: images decoded with this | |
| PiD decoder are for research/evaluation use only and are distinct in that | |
| respect from images produced by the rest of the SceneWorks pipeline. | |
| Per NVIDIA License §3.1, this distribution (a) is under the same license, | |
| (b) includes a complete copy of it, and (c) retains all copyright/patent/ | |
| trademark/attribution notices present in the original Work. NVIDIA's name, | |
| logos, and trademarks are not used except as necessary to reproduce these | |
| notices (NVIDIA License §3.5). | |