This repository is the official implementation of StreamingT2V.

StreamingT2V: Consistent, Dynamic, and Extendable Long Video Generation from Text

Roberto Henschel, Levon Khachatryan, Daniil Hayrapetyan, Hayk Poghosyan, Vahram Tadevosyan, Zhangyang Wang, Shant Navasardyan, Humphrey Shi

arXiv preprint | Video | Project page

StreamingT2V is an advanced autoregressive technique that enables the creation of long videos featuring rich motion dynamics without any stagnation. It ensures temporal consistency throughout the video, aligns closely with the descriptive text, and maintains high frame-level image quality. Our demonstrations include successful examples of videos up to 1200 frames, spanning 2 minutes, and can be extended for even longer durations. Importantly, the effectiveness of StreamingT2V is not limited by the specific Text2Video model used, indicating that improvements in base models could yield even higher-quality videos.

• [03/21/2024] Paper StreamingT2V released!
• [04/05/2024] Code and model released!

1. Clone this repository and enter:

git clone https://github.com/Picsart-AI-Research/StreamingT2V.git
cd StreamingT2V/

2. Install requirements using Python 3.10 and CUDA >= 11.6

conda create -n st2v python=3.10
conda activate st2v
pip install -r requirements.txt

3. (Optional) Install FFmpeg if it's missing on your system

conda install conda-forge::ffmpeg

4. Download the weights from HF and put them into the t2v_enhanced/checkpoints directory.

cd t2v_enhanced
python inference.py --prompt="A cat running on the street"

To use other base models add the --base_model=AnimateDiff argument. Use python inference.py --help for more options.

cd t2v_enhanced
python inference.py --image=../__assets__/demo/fish.jpg --base_model=SVD

All measurements were conducted using the NVIDIA A100 (80 GB) GPU. Randomized blending is employed when the frame count surpasses 80. For Randomized blending, the values for chunk_size and overlap_size are set to 112 and 32, respectively.

The same functionality is also available as a gradio demo

cd t2v_enhanced
python gradio_demo.py

Detailed results can be found in the Project page.

Our code is published under the CreativeML Open RAIL-M license.

We include ModelscopeT2V, AnimateDiff, SVD in the demo for research purposes and to demonstrate the flexibility of the StreamingT2V framework to include different T2V/I2V models. For commercial usage of such components, please refer to their original license.

If you use our work in your research, please cite our publication:

@article{henschel2024streamingt2v,
  title={StreamingT2V: Consistent, Dynamic, and Extendable Long Video Generation from Text},
  author={Henschel, Roberto and Khachatryan, Levon and Hayrapetyan, Daniil and Poghosyan, Hayk and Tadevosyan, Vahram and Wang, Zhangyang and Navasardyan, Shant and Shi, Humphrey},
  journal={arXiv preprint arXiv:2403.14773},
  year={2024}
}