File size: 2,258 Bytes
91758a1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | ---
license: apache-2.0
tags:
- histopathology
- diffusion
- spatial-transcriptomics
- icml-2026-sd4h-workshop
---
# STMDiT — Inference Checkpoints
EMA-only inference weights for every model row reported in the ICML 2026 SD4H
workshop submission *Transcriptomics-Conditioned Virtual Tissue Synthesis via
Diffusion Transformers*.
Each subfolder contains a single model:
- `model.pt` — EMA-only state dict (PyTorch `.pt`)
- `training_config.yaml` — the original training YAML
- `README.md` — per-model card
## Models
| `row_id` | Paper label |
|----------|-------------|
| `pixcell-b` | PixCell-B |
| `pixcell-flow-b` | PixCell-Flow-B |
| `adaln-ddpm-p01` | PixCell-GE-B (p=0.1) |
| `adaln-ddpm-p02` | PixCell-GE-B-p02 |
| `adaln-ddpm-p03` | PixCell-GE-B-p03 |
| `adaln-ddpm-p05` | PixCell-GE-B-p05 |
| `adaln-ddpm-p06` | PixCell-GE-B-p06 |
| `adaln-flow-p01` | PixCell-Flow-GE-B (p=0.1) |
| `adaln-flow-p02` | PixCell-Flow-GE-B-p02 |
| `adaln-flow-p03` | PixCell-Flow-GE-B-p03 |
| `adaln-flow-p05` | PixCell-Flow-GE-B-p05 |
| `xattn-direct-p01` | XAttn-Direct (p=0.1) |
| `xattn-gsa-p01` | XAttn-GSA (p=0.1) |
| `xattn-perceiver-p01` | XAttn-Perceiver (p=0.1) |
| `xattn-pma-p01` | XAttn-PMA (p=0.1) |
| `xattn-perceiver-p05` | XAttn-Perceiver-p05 |
| `xattn-perceiver-p06` | XAttn-Perceiver-p06 |
| `xattn-pma-p05` | XAttn-PMA-p05 |
| `xattn-pma-p06` | XAttn-PMA-p06 |
| `ptpl-adaln-p05` | PTPL-AdaLN-B (p=0.5) |
| `ptpl-adaln-p06` | PTPL-AdaLN-B-p06 |
| `ptpl-adaln-p07` | PTPL-AdaLN-B-p07 |
| `ptpl-xattn-perceiver-p05` | PTPL-XAttn-Perceiver-B (p=0.5) |
| `ptpl-xattn-perceiver-p06` | PTPL-XAttn-Perceiver-B-p06 |
| `ptpl-xattn-perceiver-p07` | PTPL-XAttn-Perceiver-B-p07 |
| `ptpl-xattn-pma-p05` | PTPL-XAttn-PMA-B (p=0.5) |
| `ptpl-xattn-pma-p06` | PTPL-XAttn-PMA-B-p06 |
| `ptpl-xattn-pma-p07` | PTPL-XAttn-PMA-B-p07 |
## Usage
```python
from huggingface_hub import snapshot_download
ckpt_dir = snapshot_download(
repo_id="stmdit-anon/stmdit-checkpoints",
allow_patterns="xattn-perceiver-p05/*", # pick one model by row_id
)
# ckpt_dir / "xattn-perceiver-p05" / "model.pt"
```
See the public anonymized code repo (linked from the OpenReview submission) for
the loader, demo notebook, and end-to-end inference example.
|