Locality Diffusion Baselines

This repository contains pretrained baseline model weights used in:

Locality in Image Diffusion Models Emerges from Data Statistics
Artem Lukoianov et al., NeurIPS 2025.

Structure

models/baseline_unet/ – baseline UNet models on multiple datasets
models/wiener/ – Wiener models

Usage

from huggingface_hub import hf_hub_download

path = hf_hub_download(
    repo_id="ottogin/locality-diffusion-baselines",
    filename="models/baseline_unet/cifar10/ckpt_epoch_200.pt"
)