File size: 1,190 Bytes
96957c2 31f9d66 96957c2 31f9d66 96957c2 31f9d66 | 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 | ---
license: apache-2.0
tags:
- world-model
- minecraft
- video-generation
- causal-forcing
---
# ForgeWM Checkpoints
Training checkpoints for [ForgeWM](https://github.com/asdfo123/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
```bash
# 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
```bibtex
@misc{forgewm2026,
title={ForgeWM: A Reproducible Training Recipe for Action-Controllable World Models},
author=ForgeWM Team,
year={2026},
url={https://github.com/asdfo123/ForgeWM}
}
```
## Contact
[Xinye Li](https://asdfo123.github.io) leeasdfo123@gmail.com |