Skip to content

Allegro is a powerful text-to-video model that generates high-quality videos up to 6 seconds at 15 FPS and 720p resolution from simple text input.

License

Notifications You must be signed in to change notification settings

rhymes-ai/Allegro

Repository files navigation

Gallery · Hugging Face · Blog · Discord · Join Waitlist (Try it on Discord!)

Allegro is a powerful text-to-video model that generates high-quality videos up to 6 seconds at 15 FPS and 720p resolution from simple text input.

Model Info

Model Allegro
Description Text-to-Video Generation Model
Download Hugging Face
Parameter VAE: 175M
DiT: 2.8B
Inference Precision VAE: FP32/TF32/BF16/FP16 (best in FP32/TF32)
DiT/T5: BF16/FP32/TF32
Context Length 79.2K
Resolution 720 x 1280
Frames 88
Video Length 6 seconds @ 15 FPS
Single GPU Memory Usage 9.3G BF16 (with cpu_offload)

Quick Start

  1. Download the Allegro GitHub code.

  2. Install the necessary requirements.

    • Ensure Python >= 3.10, PyTorch >= 2.4, CUDA >= 12.4. For details, see requirements.txt.

    • It is recommended to use Anaconda to create a new environment (Python >= 3.10) to run the following example.

  3. Download the Allegro model weights.

  4. Run inference.

    python single_inference.py \
    --user_prompt 'A seaside harbor with bright sunlight and sparkling seawater, with many boats in the water. From an aerial view, the boats vary in size and color, some moving and some stationary. Fishing boats in the water suggest that this location might be a popular spot for docking fishing boats.' \
    --vae your/path/to/vae \
    --dit your/path/to/transformer \
    --text_encoder your/path/to/text_encoder \
    --tokenizer your/path/to/tokenizer \
    --guidance_scale 7.5 \
    --num_sampling_steps 100 \
    --seed 42

Limitation

  • The model cannot render celebrities, legible text, specific locations, streets or buildings.

Future Plan

  • Multiple GPU inference and further speed up (PAB)
  • Text & Image-To-Video (TI2V) video generation
  • Motion-controlled video generation
  • Visual quality enhancement

Support

If you encounter any problems or have any suggestions, feel free to open an issue or send an email to [email protected].

License

This repo is released under the Apache 2.0 License.

Disclaimer

The Allegro series models are provided on an "AS IS" basis, and we disclaim any liability for consequences or damages arising from your use. Users are kindly advised to ensure compliance with all applicable laws and regulations. This includes, but is not limited to, prohibitions against illegal activities and the generation of content that is violent, pornographic, obscene, or otherwise deemed non-safe, inappropriate, or illegal. By using these models, you agree that we shall not be held accountable for any consequences resulting from your use.

Acknowledgment

We extend our heartfelt appreciation for the great contribution to the open-source community, especially Open-Sora-Plan, as we build our diffusion transformer (DiT) based on Open-Sora-Plan v1.2.

  • Open-Sora-Plan: A project aims to create a simple and scalable repo, to reproduce Sora.
  • Open-Sora: An initiative dedicated to efficiently producing high-quality video.
  • ColossalAI: A powerful large model parallel acceleration and optimization system.
  • VideoSys: An open-source project that provides a user-friendly and high-performance infrastructure for video generation.
  • DiT: Scalable Diffusion Models with Transformers.
  • PixArt: An open-source DiT-based text-to-image model.
  • StabilityAI VAE: A powerful image VAE model.
  • CLIP: A powerful text-image embedding model.
  • T5: A powerful text encoder.
  • Playground: A state-of-the-art open-source model in text-to-image generation.

About

Allegro is a powerful text-to-video model that generates high-quality videos up to 6 seconds at 15 FPS and 720p resolution from simple text input.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages