BonanDing commited on
Commit
5e9da5a
·
1 Parent(s): 681f346

update README

Browse files
Files changed (1) hide show
  1. README.md +18 -0
README.md CHANGED
@@ -27,6 +27,24 @@ data/
27
 
28
  The training and evaluation scripts expect the dataset to live at `data/minecraft` by default.
29
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  ## Training
31
 
32
  Run a single stage:
 
27
 
28
  The training and evaluation scripts expect the dataset to live at `data/minecraft` by default.
29
 
30
+ ## Checkpoints
31
+
32
+ Pretrained checkpoints are hosted on Hugging Face: `zeqixiao/worldmem_checkpoints`.
33
+
34
+ Example download to `checkpoints/`:
35
+
36
+ ```bash
37
+ huggingface-cli download zeqixiao/worldmem_checkpoints diffusion_only.ckpt --local-dir checkpoints
38
+ huggingface-cli download zeqixiao/worldmem_checkpoints vae_only.ckpt --local-dir checkpoints
39
+ huggingface-cli download zeqixiao/worldmem_checkpoints pose_prediction_model_only.ckpt --local-dir checkpoints
40
+ ```
41
+
42
+ Then point your scripts or configs to these files, for example:
43
+
44
+ ```bash
45
+ python -m main +name=train +diffusion_model_path=checkpoints/diffusion_only.ckpt +vae_path=checkpoints/vae_only.ckpt
46
+ ```
47
+
48
  ## Training
49
 
50
  Run a single stage: