asdfo123 commited on
Commit
96957c2
·
verified ·
1 Parent(s): 702e9c4

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +33 -0
  2. stage0/model.pt +3 -0
  3. stage3/model.pt +3 -0
README.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - world-model
5
+ - minecraft
6
+ - video-generation
7
+ - causal-forcing
8
+ ---
9
+
10
+ # ForgeWM Checkpoints
11
+
12
+ Training checkpoints for [ForgeWM](https://github.com/asdfo123/ForgeWM).
13
+
14
+ ## Available Checkpoints
15
+
16
+ | File | Stage | Description |
17
+ |------|-------|-------------|
18
+ | `stage0/model.pt` | Stage 0 | Bidirectional SFT (domain adaptation, 4000 steps) |
19
+ | `stage3/model.pt` | Stage 3 | DMD final model (4-step real-time inference) |
20
+
21
+ ## Usage
22
+
23
+ ```bash
24
+ # Download
25
+ huggingface-cli download asdfo123/ForgeWM --local-dir ./ckpts --repo-type model
26
+
27
+ # Inference
28
+ python inference.py --checkpoint_path ckpts/stage3/model.pt --image_path demo_images/forest.png --action_type forward
29
+ ```
30
+
31
+ ## 🚧 Coming Soon
32
+ - Stage 1 (Teacher-Forcing AR) checkpoint
33
+ - Stage 2 (Consistency Distillation) checkpoint
stage0/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6c825504a1e6eb05e61b25b3b2a245ff113736bdac48b1f53ec1318ae5c933a
3
+ size 7298201358
stage3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35f8fd05b38bd967804c28d96ee96f3a70662a2a4e38d48024a93813d48943ef
3
+ size 6477308409