metadata
license: apache-2.0
tags:
- world-model
- minecraft
- video-generation
- causal-forcing
ForgeWM Checkpoints
Training checkpoints for ForgeWM.
Available Checkpoints
| File | Stage | Description |
|---|---|---|
stage0/model.pt |
Stage 0 | Bidirectional SFT (domain adaptation, 4000 steps) |
stage1/model.pt |
Stage 1 | Teacher-Forcing Causal AR (10000 steps) |
stage2/model.pt |
Stage 2 | Consistency Distillation (6000 steps) |
stage3/model.pt |
Stage 3 | DMD final model (4-step real-time inference) |
Usage
# Download
huggingface-cli download asdfo123/ForgeWM --local-dir ./ckpts --repo-type model
# Inference
python inference.py --checkpoint_path ckpts/stage3/model.pt --image_path demo_images/forest.png --action_type forward
🚧 Coming Soon
- More intermediate checkpoints
Citation
@misc{forgewm2026,
title={ForgeWM: A Reproducible Training Recipe for Action-Controllable World Models},
author=ForgeWM Team,
year={2026},
url={https://github.com/asdfo123/ForgeWM}
}