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