This model is based on a multi-stage text-to-video generation diffusion model, which inputs a description text and returns a video that matches the text description. Only English input is supported.
Model description
The text-to-video generation diffusion model consists of three sub-networks: text feature extraction model, text feature-to-video latent space diffusion model, and video latent space to video visual space model. The overall model parameters are about 1.7 billion. Currently, it only supports English input. The diffusion model adopts a UNet3D structure, and implements video generation through the iterative denoising process from the pure Gaussian noise video.
You can optimize for memory usage by enabling attention and VAE slicing and using Torch 2.0.
This should allow you to generate videos up to 25 seconds on less than 16GB of GPU VRAM.
1import torch
2from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
3from diffusers.utils import export_to_video
45# load pipeline6pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")7pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)89# optimize for GPU memory10pipe.enable_model_cpu_offload()11pipe.enable_vae_slicing()1213# generate14prompt = Spiderman is surfing. Darth Vader is also surfing and following Spiderman"
15video_frames = pipe(prompt, num_inference_steps=25, num_frames=200).frames
1617# convent to video18video_path = export_to_video(video_frames)
View results
The above code will display the save path of the output video, and the current encoding format can be played with VLC player.
The output mp4 file can be viewed by VLC media player. Some other media players may not view it normally.
Model limitations and biases
The model is trained based on public data sets such as Webvid, and the generated results may have deviations related to the distribution of training data.
This model cannot achieve perfect film and television quality generation.
The model cannot generate clear text.
The model is mainly trained with English corpus and does not support other languages at the moment**.
The performance of this model needs to be improved on complex compositional generation tasks.
Misuse, Malicious Use and Excessive Use
The model was not trained to realistically represent people or events, so using it to generate such content is beyond the model's capabilities.
It is prohibited to generate content that is demeaning or harmful to people or their environment, culture, religion, etc.
Prohibited for pornographic, violent and bloody content generation.
Prohibited for error and false information generation.
Training data
The training data includes LAION5B, ImageNet, Webvid and other public datasets. Image and video filtering is performed after pre-training such as aesthetic score, watermark score, and deduplication.
(Part of this model card has been taken from here)
