Update README.md

README.md

@@ -3,3 +3,17 @@ license: other
 license_name: waymo-open-dataset-non-commercial-use
 license_link: https://waymo.com/open/terms/
 ---
+
+**Folder structure:**
+
+```bash
+<InfiniCube_repo>/checkpoints
+├── gsm_vs02_res512_view1_dual_branch_sky_mlp_modulator.ckpt # Stage 3, dual branch reconstruction model
+├── vae_epoch7_step6250.ckpt   # Stage 1, 3D voxel VAE
+├── voxel_diffusion.ckpt       # Stage 1, HD Map conditioned voxel diffusion model
+├── wan14b-i2v-buffer-step-900.safetensors  # Stage 2, 14B buffer-guided image-to-video wan2.1 model. 
+├── wan14b-t2v-buffer-step-1200.safetensors  # Stage 2, 14B buffer-guided text-to-video wan2.1 model. 
+└── wan1pt3b-t2v-buffer-step-3500.safetensors  # Stage 2, 1.3B buffer-guided image-to-video wan2.1 model (not performant)
+```
+
+In current inference script in stage 2, we only use `wan14b-t2v-buffer-step-1200.safetensors` by default.