File size: 540 Bytes
fdddca6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# 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

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