Instructions to use Yang-Tian/StereoWorld with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Yang-Tian/StereoWorld with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Yang-Tian/StereoWorld", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
| license: apache-2.0 | |
| tags: | |
| - stereo-video | |
| - world-model | |
| - diffusion | |
| # StereoWorld | |
| Official model weights for **[StereoWorld: Camera-Guided Stereo Video Generation](https://arxiv.org/abs/2603.17375)**. | |
| [](https://github.com/VAST-AI-Research/StereoWorld) | |
| [](https://arxiv.org/abs/2603.17375) | |
| ## Models | |
| This repository contains two models: | |
| | Directory | Model | Description | | |
| | --- | --- | --- | | |
| | `StereoWorldModel/` | Fixed-Baseline Stereo | Generates side-by-side stereo video with a configurable but fixed stereo baseline. Use `--use_raymap` during inference. | | |
| | `StereoWorldFlexModel/` | Flexible Stereo | Independent left/right camera control with multiple right-camera modes (converging, horizontal offset, depth offset, height offset). | | |
| ### Fixed-Baseline Stereo (`StereoWorldModel/`) | |
| A binocular teacher model that produces left-right stereo pairs with a fixed baseline between cameras. Best for standard stereo video generation where consistent disparity is desired. | |
| ### Flexible Stereo (`StereoWorldFlexModel/`) | |
| A more general multi-view world model that allows independent camera trajectories for left and right views. Supports four right-camera modes for creative stereo effects. | |
| ## Download | |
| ### Download everything | |
| ```bash | |
| huggingface-cli download Yang-Tian/StereoWorld --local-dir weights | |
| ``` | |
| ### Download only Fixed-Baseline Stereo | |
| ```bash | |
| huggingface-cli download Yang-Tian/StereoWorld \ | |
| --include "StereoWorldModel/*" \ | |
| --local-dir weights | |
| ``` | |
| ### Download only Flexible Stereo | |
| ```bash | |
| huggingface-cli download Yang-Tian/StereoWorld \ | |
| --include "StereoWorldFlexModel/*" \ | |
| --local-dir weights | |
| ``` | |
| ## Directory Structure | |
| ```text | |
| StereoWorld/ | |
| βββ StereoWorldModel/ | |
| β βββ transformer/ | |
| β βββ vae/ | |
| β βββ tokenizer/ | |
| β βββ text_encoder/ | |
| β βββ scheduler/ | |
| βββ StereoWorldFlexModel/ | |
| βββ transformer/ | |
| βββ vae/ | |
| βββ tokenizer/ | |
| βββ text_encoder/ | |
| βββ scheduler/ | |
| ``` | |
| ## Usage | |
| See the [GitHub repository](https://github.com/VAST-AI-Research/StereoWorld) for installation and inference instructions. | |
| ```bash | |
| # Fixed-Baseline Stereo | |
| python3 inference.py \ | |
| --pipeline_dir weights/StereoWorldModel \ | |
| --use_raymap \ | |
| --eval_json ExpData/demo_custom_eval.json | |
| # Flexible Stereo | |
| python3 inference_flex.py \ | |
| --pipeline_dir weights/StereoWorldFlexModel \ | |
| --eval_json ExpData/flex_demo_custom_eval.json | |
| ``` | |
| ## Citation | |
| ```bibtex | |
| @article{sun2026stereo, | |
| title={Stereo World Model: Camera-Guided Stereo Video Generation}, | |
| author={Sun Yang-Tian and Huang Zehuan and Niu Yifan and Ma Lin and Cao Yan-Pei and Ma Yuewen and Qi Xiaojuan}, | |
| journal={arXiv preprint arXiv:2603.17375}, | |
| year={2026} | |
| } | |
| ``` | |