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

Usage

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.