Allegro:强大的文本到视频模型

Allegro是一款先进的文本到视频生成模型,能够从简单的文本输入生成高质量的视频。这些视频长达6秒,以每秒15帧(FPS)的速度和720p的分辨率呈现,为视频内容创作带来了新的可能性。

Allegro模型信息

  • 模型名称:Allegro
  • 描述:文本到视频生成模型
  • 下载:可在Hugging Face上找到
  • 参数
    • VAE:175M
    • DiT:2.8B
  • 推理精度
    • VAE:FP32/TF32/BF16/FP16(推荐使用FP32/TF32)
    • DiT/T5:BF16/FP32/TF32
  • 上下文长度:79.2K
  • 分辨率:720 x 1280
  • 帧数:88
  • 视频长度:6秒 @ 15 FPS
  • 单GPU内存使用量:9.3G BF16(启用cpu_offload时)

快速开始

要开始使用Allegro,你需要按照以下步骤操作:

  1. 下载Allegro GitHub代码
  2. 安装必要的依赖
  3. 确保Python版本大于等于3.10,PyTorch版本大于等于2.4,CUDA版本大于等于12.4。具体详情可查看requirements.txt文件。
  4. 推荐使用Anaconda创建新环境(Python >= 3.10)来运行以下示例。
  5. 下载Allegro模型权重
  6. 运行推理

以下是运行推理的示例命令:

bash

复制代码
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.' \
--save_path ./output_videos/test_video.mp4 \
--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

使用--enable_cpu_offload可以将模型卸载到CPU以减少GPU内存使用(约9.3G,如果不启用CPU卸载则为27.5G),但推理时间将显著增加。

(可选)将视频插值到30 FPS。

推荐使用EMA-VFI将视频从15 FPS插值到30 FPS。

为了更好的视觉质量,请使用imageio保存视频。

限制

该模型无法渲染名人、可读文本、特定地点、街道或建筑物。

未来计划

  • 多GPU推理和进一步加速(PAB)
  • 文本&图像到视频(TI2V)视频生成
  • 动作控制视频生成
  • 视觉质量增强

结语

Allegro为视频内容创作带来了新的便利性和效率。如果你对这个模型感兴趣,可以访问其Hugging Face页面、博客、论文或加入等待名单来了解更多信息,并尝试在Discord上使用它。

相关推荐
西安老张(AIGC&ComfyUI)3 分钟前
第021章:ComfyUI文生音频Qwen3-TTS模型数字人音色设计(一)
aigc·音视频·数字人·comfyui
matlab代码6 分钟前
基于CNN卷积神经网络日常物品识别系统 (数字图像处理GUI界面)【源码37期】
人工智能·神经网络·cnn·物品识别
小殊小殊8 分钟前
一文速通GPU版FFmpeg视频转码的安装使用
ffmpeg·音视频·视频编解码
2zcode10 分钟前
基于HSV颜色空间和卷积神经网络的交通标志识别系统设计与实现
人工智能·神经网络·cnn
Σίσυφος190011 分钟前
高斯滤波 详解
人工智能
HZZD_HZZD11 分钟前
用电行为异常检测VAE-基于PyTorch设计用电行为异常检测模型:从时序特征提取到变分自编码器部署的完整实战
人工智能·pytorch·python
威视锐科技22 分钟前
AMD生态赋能5G NTN 革新:威视锐空天地一体化基站,融合天地通信与边缘AI
人工智能·5g·软件无线电·威视锐·天地一体化
库拉大叔22 分钟前
GPT内容输出优化:如何获得更符合需求的答案
人工智能
蕃茄田艺术22 分钟前
学龄儿童创意画画怎么判断是否适合自己
人工智能·蕃茄田艺术
毒爪的小新23 分钟前
踩坑实录 | RAG知识库完整搭建-Milvus2.4+BGE大中文AI模型嵌入
linux·人工智能·ai·milvus·rag