Goodeat/drifting

Model

Generative Modeling via Drifting: Pretrained Weights

2 commits

1 linked in READMEs

updated Oct 12, 2026

See the code

README

Generative Modeling via Drifting: Pretrained Weights

JAX weights for the ImageNet 256x256 models of Generative Modeling via Drifting (NeurIPS 2026, oral). Code, usage, and evaluation instructions: lambertae/drifting. Project page: lambertae.github.io/projects/drifting.

Files

Path	Description
models/gen/jax/latent_L_sota	Latent-space generator, L/2 (FID 1.54, 1-NFE)
models/gen/jax/latent_B_sota	Latent-space generator, B/2 (FID 1.75, 1-NFE)
models/gen/jax/pixel_L_sota	Pixel-space generator, L/16 (FID 1.61, 1-NFE)
models/gen/jax/pixel_B_sota	Pixel-space generator, B/16 (FID 1.76, 1-NFE)
models/gen/jax/ablation	Ablation-setting generator, B/2
models/mae/jax/mae_latent_640	Latent-space MAE-640 feature encoder
models/mae/jax/mae_pixel_640	Pixel-space MAE-640 feature encoder
models/mae/jax/mae_latent_256	Latent-space MAE-256 feature encoder (ablations)

The code downloads these weights automatically.

License

MIT, same as the code repository.

Citation

Please cite the paper; the BibTeX entry is in the code repository.

image-generation

one-step-generation

Similar repos

lambertae/drifting

576

hustvl/lightningdit-xl-imagenet256-64ep

Model

This model is introduced in the paper Reconstruction vs. Generation: Taming Optimization Dilemma in…

nvidia/PixelDiT-ImageNet

Model

Bili-Sakura/Visual-Generative-Foundation-Model-Collection

A collection of visual generative foundation model including code, paper, checkpoint etc.

jjiaweiyang/FD-Loss

Model

Representation Fréchet Loss for Visual Generation

Lyy-iiis/pMF

Official Implementation of pMF https://arxiv.org/abs/2601.22158

296

chenguolin/DiffSplat

Model

[[ICLR 2025] DiffSplat](https://chenguolin.github.io/projects/DiffSplat)

zhen-nan/L2P

Model

L2P: Unlocking Latent Potential for Pixel Generation

Linked in READMEs

lambertae/drifting

576

README badge

Goodeat/drifting

Model

Generative Modeling via Drifting: Pretrained Weights

2 commits

1 linked in READMEs

updated Oct 12, 2026

See the code

README

Generative Modeling via Drifting: Pretrained Weights

Files

Path	Description
models/gen/jax/latent_L_sota	Latent-space generator, L/2 (FID 1.54, 1-NFE)
models/gen/jax/latent_B_sota	Latent-space generator, B/2 (FID 1.75, 1-NFE)
models/gen/jax/pixel_L_sota	Pixel-space generator, L/16 (FID 1.61, 1-NFE)
models/gen/jax/pixel_B_sota	Pixel-space generator, B/16 (FID 1.76, 1-NFE)
models/gen/jax/ablation	Ablation-setting generator, B/2
models/mae/jax/mae_latent_640	Latent-space MAE-640 feature encoder
models/mae/jax/mae_pixel_640	Pixel-space MAE-640 feature encoder
models/mae/jax/mae_latent_256	Latent-space MAE-256 feature encoder (ablations)