ForgeWM Checkpoints

Training checkpoints for ForgeWM.

Available Checkpoints

File	Stage	Description
`stage0/model.pt`	Stage 0	Bidirectional SFT (domain adaptation, 4000 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

Stage 1 (Teacher-Forcing AR) checkpoint
Stage 2 (Consistency Distillation) checkpoint

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support