Instructions to use acvlab/ABot-World-0-5B-LF with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use acvlab/ABot-World-0-5B-LF with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image, export_to_video # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("acvlab/ABot-World-0-5B-LF", dtype=torch.bfloat16, device_map="cuda") pipe.to("cuda") prompt = "A man with short gray hair plays a red electric guitar." image = load_image( "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png" ) output = pipe(image=image, prompt=prompt).frames[0] export_to_video(output, "output.mp4") - Notebooks
- Google Colab
- Kaggle
🌍 ABot-World: Infinite Interactive World Rollout on a Single Desktop GPU
TL;DR: ABot-World turns a single NVIDIA RTX 5090 desktop GPU into a real-time interactive world simulator, enabling infinite action-conditioned world rollout at 720P, 16 FPS, 1.2s latency, and 19GB GPU memory.
🚀 Key Highlights
- 🎮 Action-Driven World Control: Responds to user actions in real time, enabling continuous exploration instead of passive video playback.
- ⚡ Real-Time Desktop Inference: Runs at 720p and 16 FPS on a single NVIDIA RTX 5090 desktop GPU, with 1.2s latency and 19GB GPU memory.
- ♾️ Infinite World Rollout: Supports open-ended interactive world generation beyond fixed video-length limits.
- 🧠 Open-Ended World Imagination: Expands the world with new scenes and dynamics during rollout, avoiding scene lock-in, without prompt switching, by our LongForcing training.
📢 News
- 2026-08-03: We release the dataset of
500-hourtraining data. - 2026-07-22: Release ABot-World-0 technical report.
- 2026-07-13: ABot-World is now on Reactor!
- 2026-07-10: We have decided to open-source our
500-hourvideo training dataset with accurate action annotations. Stay tuned—we plan to release it very soon. - 2026-07-09: We release the causal student model
ABot-World-0-5B-LF, inference code, our local gradio demo and online playground ABot World Studio.
🛠️ Setup
This installation was tested on: Ubuntu 22.04, CUDA 13.3, NVIDIA RTX 5090.
- Clone the repository:
git clone https://github.com/amap-cvlab/ABot-World.git
cd ABot-World
- Install dependencies using conda:
conda create -n aworld python=3.12 -y
conda activate aworld
pip install -r requirements.txt
- Download checkpoints:
Download models using HuggingFace:
pip install -U "huggingface_hub"
hf download acvlab/ABot-World-0-5B-LF --local-dir ./checkpoints/ABot-World-0-5B-LF
Download models using ModelScope:
pip install -U "modelscope"
modelscope download "amap_cvlab/ABot-World-0-5B-LF" --local_dir ./checkpoints/ABot-World-0-5B-LF
After downloading, the project should have the following checkpoint structure:
checkpoints/
└── ABot-World-0-5B-LF/
├── Wan2.2_VAE.pth
├── taew2_2.pth
├── models_t5_umt5-xxl-enc-bf16.pth
├── diffusion_pytorch_model.safetensors
└── google/umt5-xxl/
The checkpoint paths are configured in configs/long_forcing_dmd.yaml and
configs/default_config.yaml. The distilled generator weights are already
merged into ABot-World-0-5B-LF/diffusion_pytorch_model.safetensors.
🤗 Gradio Demo
bash web_client/run.sh
Select a GPU with:
CUDA_ID=0 bash web_client/run.sh
License
This project is released under the Apache License 2.0. See LICENSE, NOTICE,
and THIRD_PARTY_NOTICES.md for copyright and third-party attribution details.
🤝 Acknowledgement
This project builds on and is inspired by the following open-source projects: Causal Forcing, AngelSlim, LightX2V, taehv, Wan2.2, Helios, from which the optimized Triton RoPE and normalization kernels in wan/modules/helios_kernels are derived.
🗓️ Roadmap
- Interactive Web Playground (ABot World Studio)
- Inference Code Release
- Local Gradio Demo Release
- Causal Student Model Release
- Technical Report (arXiv)
- Bidirectional Teacher Model Release
- 500-Hour Video Training Dataset with Accurate Action Annotations
📝 Citation
If you find our work helpful, please cite our paper:
@misc{jiang2026abotworld0,
title={{ABot-World-0}: Infinite Interactive World Rollout on a Single Desktop GPU},
author={Fan Jiang and Zhaoxu Sun and Mengchao Wang and Ziyu Zhu and Chiyu Wang and Yunpeng Zhang and Wenlin Liu and Yun Wang and Xue Zheng and Rui Sun and Junfeng Ni and Hongyu Pan and Zhongxu Sun and Fei Yu and Zengye Ge and Mengmeng Du and Nianfei Fan and Mingchao Sun and Yu Liu and Yongchang and Yanqing Zhu and Jiahang Wang and Ning Ying and Yuze Xuan and Di Yang and Zhicheng Liu and Zhe Gao and Tingbing Xu and Jiacheng Sui and Wenjin Yang and Junnan Lai and Shufeng Liu and Yuan Liu and Zheng Zhou and Yingliang Peng and Dawei Cao and Kaifeng Sheng and Yuxiang Cai and Fei Lu and Mu Xu and Ning Guo},
year={2026},
eprint={2607.19191},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2607.19191},
}
🛰 Contact Us via WeChat Group
Feel free to contact us!
- Downloads last month
- 1,201
Model tree for acvlab/ABot-World-0-5B-LF
Base model
Wan-AI/Wan2.2-TI2V-5B