| --- |
| 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. |
|
|
