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