中文

Introduction to Helios

The official implementation of Helios, with released training and inference code and weights.

Verified

Core facts

Read by a script from the official API, not rewritten by a model.

License
Apache-2.0
Languages
Python
Maintainer
PKU-YuanGroup
Verified

What it is

This repository is the official implementation of Helios. The authors have released the training and inference code and weights of Helios-Base, Helios-Mid and Helios-Distilled. HeliosBench is a specialized benchmark for real-time long-video generation. This project is released under the Apache 2.0 license.

Main features

  1. Helios supports various Context Parallelism mechanisms, including Ulysses Attention, Ring Attention, Unified Attention, and Ulysses Anything Attention.

Getting started

  1. Models can be downloaded using huggingface-cli.

  2. Models can also be downloaded using modelscope-cli.

From the README · bash
# 0. Clone the repo
git clone --depth=1 https://github.com/PKU-YuanGroup/Helios.git
cd Helios

# 1. Create conda environment
conda create -n helios python=3.11.2
conda activate helios

# 2. Install PyTorch (adjust for your CUDA version)
# CUDA 12.6
pip install torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 --index-url https://download.pytorch.org/whl/cu126
# CUDA 12.8
pip install torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 --index-url https://download.pytorch.org/whl/cu128
# CUDA 13.0
pip install torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 --index-url https://download.pytorch.org/whl/cu130

# 3. Install dependencies
bash install.sh
From the README · sh
pip install "huggingface_hub[cli]"
huggingface-cli download BestWishYSH/Helios-Base --local-dir BestWishYSH/Helios-Base
huggingface-cli download BestWishYSH/Helios-Mid --local-dir BestWishYSH/Helios-Mid
huggingface-cli download BestWishYSH/Helios-Distilled --local-dir BestWishYSH/Helios-Distilled
From the README · sh
pip install modelscope
modelscope download BestWishYSH/Helios-Base --local_dir BestWishYSH/Helios-Base
modelscope download BestWishYSH/Helios-Mid --local_dir BestWishYSH/Helios-Mid
modelscope download BestWishYSH/Helios-Distilled --local_dir BestWishYSH/Helios-Distilled
From the README · bash
cd scripts/inference

# For Helios-Base
bash helios-base_t2v.sh
bash helios-base_i2v.sh
bash helios-base_v2v.sh

# For Helios-Mid
bash helios-mid_t2v.sh
bash helios-mid_i2v.sh
bash helios-mid_v2v.sh

# For Helios-Distilled
bash helios-distilled_t2v.sh
bash helios-distilled_i2v.sh
bash helios-distilled_v2v.sh

# For Interactive
# ⚠️ This feature is still under development — results may not always meet expectations
cd scripts/inference/experiment_interactive
From the README · bash
pip install git+https://github.com/huggingface/diffusers.git
From the README · bash
pip install git+https://github.com/vllm-project/vllm-omni.git

Who it is for

Helios uses an autoregressive approach that generates 33 frames per chunk. The project advises setting num_frames to a multiple of 33. If a non-multiple value is provided, it is automatically rounded up to the nearest multiple of 33.

The three models share the same architecture, while Helios-Mid and Helios-Distilled use a more aggressive multi-scale sampling pipeline to achieve better efficiency. Helios-Mid is an intermediate checkpoint generated while distilling Helios-Base into Helios-Distilled, and may not meet expected quality. For Image-to-Video or Video-to-Video, since training is based on Text-to-Video, these two functions may be slightly inferior to Text-to-Video.

Real-time generation performance depends not only on the GPU, but also on the CPU, memory, CUDA driver version, etc. Training uses a three-stage progressive pipeline. Stage-1 (Base) performs architectural adaptation, applying Unified History Injection, Easy Anti-Drifting, and Multi-Term Memory Patchification to convert the bidirectional pretrained model into an autoregressive generator.

Stage-2 (Mid) targets token compression by introducing Pyramid Unified Predictor Corrector, which aggressively reduces the number of noisy tokens and thus the overall computation. Stage-3 (Distilled) applies Adversarial Hierarchical Distillation, reducing the sampling steps from 50 to 3 and eliminating the need for classifier-free guidance (CFG). Throughout training, dynamic shifting is applied to all timestep-dependent operations to match the noise schedule to the latent size.

CNAPS.AI has provided a flexible workflow of Helios on their online inference service platform. After training, a script can merge all the checkpoints to obtain the final safetensors file.

Sources

This page is written from the official sources below. Open them to check the original text.

Verified ·

This page is written by AI from the official sources above and checked automatically. This site has not tested it. Found a mistake? Tell us.