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
File size: 2,881 Bytes
7720487 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | ---
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}
}
```
|