在 TPU 上进行推理,而不是在 GPU 上。

大家好! 我们的 AI 超级计算机团队将 Flux 推理实现移植到了 MaxDiffusion,并成功地使用 Google 的 TPU 运行了 Flux-dev 和 Flux-schnell 模型。