yifanlu/InfiniCube-checkpoints

Folder structure:

<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.

---

License

These weights are subject to the "Waymo Dataset License Agreement for Non-Commercial Use (March 2025)". By downloading or using these weights, you agree to comply with Waymo's terms, which include, but are not limited to:

• Use for Non-Commercial purposes only.
• No redistribution of the weights except as permitted by Waymo's terms.
• Compliance with all other restrictions outlined at: https://waymo.com/open/terms/