跳到正文
原文
Google Developers Blog·· 4 天前AI 評分52

TPU 上影片擴散時空注意力的加速實踐

Accelerating Spatio-Temporal Attention for Video Diffusion on TPUs

AI 導讀

Google 團隊在 TPU v6e 上優化 SVG 稀疏注意力的執行流程,令 720p、81 幀影片的端到端去噪提速 1.28×。去噪時間由 153.50 s 降至 119.86 s;在 1440p(302K tokens)下則由 2,471 s 降至 1,461 s,提速 1.69×,PSNR 為 24.05 dB。

來源:Google Developers Blog · developers.googleblog.com