Modelpublic

ottogin/locality-diffusion-baselines

sourceHugging Faceupdated 7mo agoView on Hugging Face

1likes

Model Card

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.

Please consult this [repo](https://github.com/ottogin/locality-in-diffusion-models) for the full code

Structure

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

Usage

python

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"
)