SolarWM README

The project's README from github.com/Junchao-cs/SolarWM. As one markdown file: README.md.

SolarWM: Open Data and Scalable Training for Long-Horizon Video World Models

Junchao Huang1,2   Guian Fang3   Shengju Qian4   Xianghao Kong5   Zhuoran Zhao5,6
Wei Huang7   Yihua Du6   Zixin Zhang6   Justin Cui8   Yuchao Gu7   Yukang Chen7   Xinting Hu
Tianyu He9   Shaoshuai Shi   Zhuotao Tian2   Xin Wang   Mike Zheng Shou3   Li Jiang1,2
1CUHK-SZ   2SLAI   3NUS   4CUHK   5HKUST   6HKUST-GZ   7NVIDIA   8UCLA   9MSRA

Project page Paper Hugging Face dataset ModelScope dataset Dataset Access Form Model weights Apache 2.0 license

SolarWM teaser: one framework for diverse interactive worlds

We present SolarWM, a fully open foundation for building interactive video world models from data preparation through scalable training and long-horizon inference.

News

Training progression

The staged workflow is Stage0.5 FM → Stage1 TF-AnyFlow → Stage2 DMD via SGF, turning a bidirectional video model into a camera-controlled few-step autoregressive model.

Backend Stage0.5 (Bid-Cam) Stage1 (TF-AnyFlow) Stage2 (SGF) Runtime interfaces
Wan2.2-5B ✓ ✓ ✓ train, infer, preencode
Wan2.2-14B ✓ Coming soon Coming soon train, infer, preencode
LTX-2.5 ✓ Coming soon Coming soon train, infer, preencode
MiniMax-H3 ✓ Coming soon Coming soon train, infer, preencode

Install

Wan, LTX, and MiniMax-H3 require separate runtime environments. Activate the environment for the selected backbone, then install the shared SolarWM source:

python -m pip install -e .
solarwm environment probe

Model weights are available from the SolarWM model collection. See the Wan2.2 TI2V-5B guide for Stage0.5, Stage1, and Stage2 commands. Data access options are described below.

Data

The public SolarWM-Data release contains release controls, licenses, recipe and test indexes, small format examples, and the SolarWM-Data-Annotation/ package. It does not include the full releases-v1/raw-wds/ or releases-v1/latent-wds/ payloads.

For a released recipe that uses preencoded data, download its matching latent generation. That is sufficient for training and does not require raw-WDS. Raw-WDS is needed only when you want the full processed video corpus, online encoding, your own latent generation, or another workflow whose index points to raw data.

  1. Use preencoded latents. Each latent generation is published in a separate repository. See the latent-WDS release list for available downloads and generations that are still being uploaded.
  2. Rebuild raw-WDS from annotations. SolarWM-Data-Annotation/ is an annotation-only release with no videos. It contains the released camera trajectories, captions, metadata, source identities, and reconstruction tools. Follow its README to download the original videos from their source publishers and process them into the expected raw-wds/ layout.
  3. Request prepared raw-WDS. Submit the Dataset Access Form. Approved applicants receive download instructions by email.

See the dataset access guide for download commands and the payload required by each training, validation, and inference example.

Unified commands

# Validate and render the exact resolved configuration.
solarwm config resolve \
  --config configs/examples/wan22_ti2v_5b/train_stage0p5_fm_153f.yaml \
  --set model.base_path=/path/to/SolarWM-models/SolarWM-5B-base \
  --set checkpoint.path=/path/to/SolarWM-models/SolarWM-5B-bid-stage0p5-81f/model.pt \
  --set data.index_root=/path/to/SolarWM-Data/releases-v1 \
  --set data.transport.root=/path/to/SolarWM-Data/releases-v1 \
  --set runtime.validate_every=0 \
  --set runtime.output_dir=/path/to/output

# Train the released preencoded recipe without raw-WDS.
torchrun --standalone --nproc-per-node=8 \
  -m solarwm train \
  --config configs/examples/wan22_ti2v_5b/train_stage0p5_fm_153f.yaml \
  --set distributed.world_size=8 \
  --set train.global_batch_size=8 \
  --set model.base_path=/path/to/SolarWM-models/SolarWM-5B-base \
  --set checkpoint.path=/path/to/SolarWM-models/SolarWM-5B-bid-stage0p5-81f/model.pt \
  --set data.index_root=/path/to/SolarWM-Data/releases-v1 \
  --set data.transport.root=/path/to/SolarWM-Data/releases-v1 \
  --set runtime.validate_every=0 \
  --set runtime.output_dir=/path/to/output

# Run Wan2.2-5B Stage2 inference for the longest camera-backed horizon.
torchrun --standalone --nproc-per-node=1 -m solarwm infer \
  --config configs/examples/wan22_ti2v_5b/infer_stage2_sgf_camera_length.yaml \
  --set model.base_path=/path/to/SolarWM-models/SolarWM-5B-base \
  --set checkpoint.path=/path/to/SolarWM-models/SolarWM-5B-sgf-stage2-81f \
  --set data.index_root=/path/to/SolarWM-Data/releases-v1 \
  --set data.transport.root=/path/to/SolarWM-Data/releases-v1 \
  --set inference.run_id=my-camera-run \
  --set runtime.output_dir=/path/to/output

# Preencode raw-WDS for the Wan2.2-5B 153f recipe.
torchrun --standalone --nproc-per-node=8 \
  -m solarwm preencode \
  --config configs/examples/wan22_ti2v_5b/preencode_153f.yaml \
  --set model.base_path=/path/to/SolarWM-models/SolarWM-5B-base \
  --set data.index_root=/path/to/wan153f-fixed-window-index \
  --set data.transport.root=/path/to/SolarWM-Data/releases-v1 \
  --set preencode.output_root=/path/to/latent-wds/wan22-ti2v5b-153f-480p-v1 \
  --set preencode.logical_output_root=/path/to/recipes/wan22-ti2v5b-153f-480p-v1 \
  --set runtime.output_dir=/path/to/output

The quickstart walks through a complete Wan2.2-5B setup. Copyable commands for every released route are in the backend guides.

Every launch writes resolved-config.json and launch-manifest.json before model allocation. Config overrides are explicit and included in the resolved configuration identity.

Local and bucket data

Index rows always contain POSIX shard keys relative to the release directory:

{"sample_id":"...","shard":"raw-wds/abot/shards/kept-high-000001.tar"}

Only the runtime root changes:

# Locally mounted storage
data:
  index_root: /path/to/SolarWM-Data/releases-v1
  transport:
    kind: local
    root: /path/to/SolarWM-Data/releases-v1

# Object-store streaming uses the same release-relative rows. Supply the
# release root from the distribution channel or deployment environment.
data:
  index_root: /path/to/SolarWM-Data/releases-v1
  transport:
    kind: gcs
    root: ${SOLAR_RELEASE_ROOT}
    cache_dir: /path/to/solar-cache
    cache_max_gib: 256

Documentation

Acknowledgements

We gratefully thank the teams behind Wan2.2, LTX-2.5, and MiniMax-H3 for releasing the code and pretrained models that make the SolarWM backbone family possible. We also thank the authors of all datasets and open-source projects listed in NOTICE.

Citation

If you use SolarWM-Data, the data engine, or the released models in your research, please cite our paper.

Paper: https://arxiv.org/abs/2609.02886

@misc{huang2026solarwmopendatascalable,
      title={SolarWM: Open Data and Scalable Training for Long-Horizon Video World Models}, 
      author={Junchao Huang and Guian Fang and Shengju Qian and Xianghao Kong and Zhuoran Zhao and Wei Huang and Yihua Du and Zixin Zhang and Justin Cui and Yuchao Gu and Yukang Chen and Xinting Hu and Tianyu He and Shaoshuai Shi and Zhuotao Tian and Xin Wang and Mike Zheng Shou and Li Jiang},
      year={2026},
      eprint={2609.02886},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2609.02886}, 
}

License and attribution

SolarWM is licensed under Apache-2.0. See LICENSE and NOTICE. Model weights and backbone packages may carry their own licenses; review the license and model card in the corresponding release package before use or redistribution. In particular, LTX-2.5 derivatives are subject to the LTX-2.x Community License, and the MiniMax H3 Community License contains territory restrictions. Those packages are not relicensed under the SolarWM code license.

SolarWM (Junchao-cs/SolarWM)

Searching Database…

这是什么?SolarWM 是一个完全开放的构建交互式视频世界模型的基础框架,涵盖数据准备、可扩展的多主干训练和长时程推理。
这是单个模型,还是完整的研究流水线?它是一套完整的流水线/框架:一个训练栈封装了四种不同的扩散视频主干(Wan2.2、LTX-2.5、MiniMax-H3)并保留各自原生的表示和训练目标,还配有配套的数据引擎。
怎么让模型开始训练或运行?从 Wan2.2-5B 快速上手开始:激活该主干自己的运行环境,在其上安装共享的 SolarWM 源码,然后按照后端指南执行 Stage0.5/Stage1/Stage2 命令。
使用它之前需要安装什么?Wan2.2、LTX-2.5 和 MiniMax-H3 各自需要单独的运行环境;你要激活所选主干的环境,然后在其上安装共享的 SolarWM 源码包。
我可以直接下载完整数据集吗?不能。公开发布的只有元数据、索引和一个仅含标注的包;不包含完整的原始视频分片和潜在载荷,因此你要么使用预编码的 latents,要么自己从标注重建原始数据,要么通过访问申请表索取准备好的 raw-WDS 副本。
模型权重和代码使用相同的许可证吗?SolarWM 代码采用 Apache-2.0 许可,但 LTX-2.5 的衍生内容受 LTX-2.x Community License 约束,MiniMax H3 权重则带有 MiniMax H3 Community License,其中包含地区限制,且不会按 SolarWM 代码许可证重新授权。
训练这些模型需要强大的 GPU 算力吗?该框架通过多阶段扩散流水线训练 5B 到 33B 参数的模型,这意味着需要相当可观的多 GPU 算力,不过 README 并没有给出确切的硬件规格。
SolarWM 使用的三阶段训练方案是什么?Stage0.5 训练整段双向流匹配以构建基础表示;Stage1 把教师强制与 AnyFlow 损失结合起来,让模型同时学习去噪和有限步流映射;Stage2 通过自梯度强制(DMD/SGF)在冻结教师和可训练评论家的对抗下蒸馏因果学生模型。
它如何仅凭 5 秒的训练片段获得数分钟到数小时的推演?仅用 5 秒序列训练后,得到的因果模型就能支持跨越数分钟到数小时的实时交互推演,无需长序列微调或注意力汇聚(attention-sink)技巧。
SolarWM 目前支持哪些视频主干?四个模型,参数量从 5B 到 33B,基于 Wan2.2、LTX-2.5 和 MiniMax-H3 构建,每个都在共享框架内保留各自原生的表示和训练目标。
发布的数据集有多大?结构是怎样的?SolarWM-Data 把来自 14 个源数据集的 143 万个规范片段统一到一个帧对齐的契约中,涵盖观测、度量相机几何、字幕说明、质量元数据、选取和来源信息。
Creator
Junchao-cs
Size
12.5 MB
Stars
778
Updated
CommitOct 16
Age
1mo
Commits · 90 Days