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