跳到正文
原文
Google Developers · AI·· 59 分钟前精选AI 评分67

HeyGen 将 Avatar IV 迁移至 TPU,性能提升 1.86 倍

HeyGen x Google Cloud: Bringing Avatar IV to TPUs

AI 导读

HeyGen 与 Google Cloud 合作将 Avatar IV 扩散管线迁移至八芯片 Trillium (v6e) TPU 主机,比首版快 1.86 倍,同时保持输出质量一致。优化覆盖自定义注意力内核、Ulysses 序列并行、稀疏注意力去掩码、softmax 依赖解链与编译器契约,并设有字节级与相似度双重质量门。

推荐理由

原文给出了具体加速比与成本变化,读者可据此判断 TPU 部署对视频生成工作流的实际影响。

来源:Google Developers · AI · developers.googleblog.com