---
title: PAE-Collections
canonical_url: "https://www.modelscope.cn/models/ZhengrongYue/PAE-Collections"
md_url: "https://www.modelscope.cn/models/ZhengrongYue/PAE-Collections.md"
repository: ZhengrongYue/PAE-Collections
last_updated: 2026-05-08
pipeline_tag: text-to-image-synthesis
tasks:
  - text-to-image-synthesis
library_name:
  - pytorch
frameworks:
  - Pytorch
downloads: 8
stars: 0
---

# PAE-Collections

> PAE-Collections - ZhengrongYue 在 ModelScope 开源的模型。PAE Model Collections

- **Repository**: ZhengrongYue/PAE-Collections
- **Tasks**: text-to-image-synthesis
- **Downloads**: 8
- **Stars**: 0
- **Last updated**: 2026-05-08

Source: https://www.modelscope.cn/models/ZhengrongYue/PAE-Collections

---

# PAE Model Collections

## Repository Structure

```
PAE-Collections/
├── AE-models/                    # Inference-only autoencoder weights (model state_dict only)
│   ├── dinov2-large.pt           #   PAE with DINOv2-Large encoder
│   ├── dinov3-large.pt           #   PAE with DINOv3-Large encoder
│   ├── siglip2-so400m.pt         #   PAE with SigLIP2-SO400M encoder
│   └── mae-large.pt              #   PAE with MAE-Large encoder
│
├── DiT-models/                   # LightningDiT generator checkpoints
│
├── dinov2-large/                 # Full training checkpoint (DINOv2-Large)
│   └── ep-last.pt
├── dinov3-large/                 # Full training checkpoint (DINOv3-Large)
│   └── ep-last.pt
├── siglip2-so400m/               # Full training checkpoint (SigLIP2-SO400M)
│   └── ep-last.pt
└── mae-large/                    # Full training checkpoint (MAE-Large)
    └── ep-last.pt
```

## Loading Weights

```python
import torch

# Inference-only (recommended)
ckpt = torch.load("AE-models/dinov2-large.pt", map_location="cpu")
model.load_state_dict(ckpt["model"])
ckpt = torch.load("dinov2-large/ep-last.pt", map_location="cpu")
model.load_state_dict(ckpt["model"])
```