Project Icon

animatediff-cli

命令行工具实现低内存消耗的无限长度动画生成

animatediff-cli是一款开源的动画生成工具,专注于降低VRAM使用并支持无限长度动画创作。该工具兼容CPU和GPU运行,集成了RIFE插值技术以提升动画质量。用户可通过灵活的配置选项自定义提示、选择模型和设置帧数。此外,animatediff-cli支持批量生成和Python程序调用,为动画创作提供了多样化的应用场景。尽管项目仍在持续优化中,但其已展现出生成高质量动画的潜力。

animatediff

pre-commit.ci 状态

animatediff 重构,显著降低了显存使用量。

同时,**支持无限生成长度!**太棒了!

LoRA 加载功能目前绝对没有实现!

欢迎提交 PR!😆😅

理论上可以在 CPU 上运行,但不推荐。应该可以在 GPU 上正常运行,无论是 NVIDIA 还是其他品牌,但我还没有在非 CUDA 硬件上测试过。默认使用 PyTorch 2.0 缩放点积注意力(即内置 xformers),但如果你真的想使用 xformers,可以传递 --xformers 参数强制使用。

如何使用

我应该写一些更详细的步骤,但以下是大致流程:

git clone https://github.com/neggles/animatediff-cli
cd animatediff-cli
python3.10 -m venv .venv
source .venv/bin/activate
# 安装 Torch。使用你喜欢的 torch 版本(>= 2.0.0),但在非 NVIDIA 设备上可能会遇到困难...
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# 安装其余所有依赖(应该是全部了!我可能漏掉了一些依赖。)
python -m pip install -e '.[dev]'
# 现在你应该可以运行
animatediff --help
# 这里会打印出一个漂亮的帮助屏幕,里面有很多信息。

接下来,你需要将想要使用的检查点放入 data/models/sd 目录,复制 config/prompts 中的一个提示配置文件,编辑它以设置你选择的提示和模型(提示 .json 文件中的模型路径是相对于 data/,例如 models/sd/vanilla.safetensors),然后就可以开始了。

对于 8GB 显存的显卡,可以使用类似以下的命令:

animatediff generate -c 'config/prompts/waifu.json' -W 576 -H 576 -L 128 -C 16

对于显存少于 8GB 的显卡,你可能需要将 -C 降低到 8,而对于显存更大的显卡,你可以将其提高到 20-24。最大值为 24。

注意:生成 128 帧会非常慢...

RiFE!

我已经为 rife-ncnn-vulkan 添加了实验性支持,使用 animatediff rife interpolate 命令。它有相当直观的帮助信息,并已在 Linux 上进行了测试,但我不确定它是否能在 Windows 上运行。

无论如何,你需要在系统上安装 ffmpeg 并将其添加到 PATH 中,还需要从 GitHub 仓库(上面的链接)下载适用于你所选操作系统的 rife-ncnn-vulkan 发布版。解压它,并将提取的文件夹放在 data/rife/ 目录下。你应该有一个 data/rife/rife-ncnn-vulkan 可执行文件,或者在 Windows 上是 data\rife\rife-ncnn-vulkan.exe

你还需要使用以下命令重新安装仓库/包:

python -m pip install -e '.[rife]'

或者手动安装 ffmpeg-python

默认设置是将每帧乘以 8,将 8fps 的动画转换为 64fps,然后编码为 60fps 的 WebM。(如果你选择 GIF 模式,它将是 50fps,因为 GIF 的帧持续时间以 1/100 秒为单位编码,这很诡异)。

看起来效果还不错...

待办事项:

无特定顺序:

  • 支持无限生成长度
  • 支持 RIFE 进行运动插值(rife-ncnn-vulkan 不是最佳实现)
  • 将 RIFE 插值后的帧导出为视频文件(webm、mp4、动画 webp、hevc mp4、gif 等)
  • 在 6-8GB 显存的显卡上生成无限长度的动画(512x512 分辨率,8 帧上下文,但至少能用)
  • Torch SDP 注意力(使 xformers 成为可选项)
  • 在提示配置中支持 clip_skip
  • 实验性支持 torch.compile()(上游 Diffusers 的 bug 略微减慢了速度,但仍然很快)
  • 使用 --repeat 批量生成!(例如,--repeat 10 将重复所有提示 10 次)
  • 从另一个 Python 程序调用 animatediff.cli.generate() 函数,无需每次重新加载模型
  • 将剩余的旧 Diffusers 代码更新到最新版本(大部分已完成)
  • 添加 Web 界面(也许吧,已经有人在封装这个项目了,所以可能不需要)
  • img2img 支持(从现有图像开始并继续生成)
  • 尽可能停止使用自定义模块(应该能够使用 Diffusers 完成几乎所有功能)
  • 自动生成然后使用 RIFE 插值的模式

致谢:

参见 guoyww/AnimateDiff(这里很少有我的工作)

注意:COPYING 文件中的版权声明缺少原作者的名字,仅仅是因为原始仓库(在撰写本文时)没有在许可证中附加姓名。然而,我使用了与他们相同的许可证(Apache 2.0)。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号