链氪巴士
  1. 首页
  2. 站点导航
  3. 文章
  4. 快讯
  5. 软件
  6. 书刊
  7. 关于

CogVideo-文本与图像到视频生成项目

CogVideo是一个专注于文本和图像到视频生成的项目,包含CogVideoX(2024)和CogVideo(ICLR2023)相关内容。该项目有多个版本的模型,不同模型有着不同的特性,如在不同的GPU上内存使用量、推理速度、支持的分辨率和推理精度等方面存在差异。项目不断更新,...

访问官网 →

项目标题:CogVideo-开源的文本与图像到视频生成项目

项目简介

CogVideo是一个有关文本和图像到视频生成的项目,包含CogVideoX(2024)和CogVideo(ICLR2023)。该项目有多种版本的模型,如CogVideoX-5B、CogVideoX-2B等,这些模型支持不同的分辨率、推理精度,在不同的GPU上有着不同的内存使用情况和推理速度。项目不断更新,例如2024年11月15日发布了CogVideoX1.5模型的diffusers版本,并且仅需进行小参数调整就能继续使用之前的代码。

核心技术

从项目中可知其模型训练与推理涉及到多种技术参数的运用。如在推理精度方面,支持BF16(推荐)、FP16、FP32、FP8、INT8等多种精度。在不同模型如CogVideoX-2B和CogVideoX-5B的训练中,分别采用了FP16精度和BF16精度训练。同时,项目提到可以利用PytorchAO和Optimum-quanto对文本编码器、变压器和VAE模块进行量化以减少内存需求,并且TorchAO量化与torch.compile完全兼容,可以显著提高推理速度。

实践案例

项目中有许多基于CogVideoX架构的修改管道或相关应用。例如CogVideoX-Fun,它支持灵活的分辨率和多种启动方法;CogStudio为CogVideo的Gradio WebUI提供了更多功能的Web界面;XorbitsInference是一个强大的分布式推理框架,可以轻松部署模型;还有ComfyUI-CogVideoXWrapper可将CogVideoX集成到工作流程中等等。另外还提供了一些可直接在免费Colab T4实例上运行的项目代码,像CogVideoX-5B-T2V-Colab.ipynb等,方便开发者进行文本到视频、量化文本到视频推理、图像到视频、视频到视频等操作。

项目总结

CogVideo项目在文本与图像到视频生成领域有着丰富的成果和不断的发展,其众多的模型版本和配套的工具、应用为开发者和使用者提供了多种选择。欢迎大家在评论区留言,分享自己对这个项目的看法或者使用过程中的经验。

项目地址

GitHub链接直达