在 MacBook (M2) 上运行由 Stable Diffusion 原始开发者推出的 FLUX.1 图像生成 AI 模型 ([dev]/[schnell])

💡 原文英文,约1400词,阅读约需6分钟。
📝

内容提要

黑森林实验室推出了开源图像生成AI模型Flux,参数规模为120亿。Flux有三种版本:FLUX.1 [pro]、FLUX.1 [dev]和FLUX.1 [schnell]。本文介绍了如何在MacBook上运行FLUX.1 [dev],并解决内存和数据类型问题。FLUX.1 [schnell]速度更快,但质量稍逊。实验室计划推出文本到视频系统。

🎯

关键要点

  • 黑森林实验室推出了开源图像生成AI模型Flux,参数规模为120亿。
  • Flux有三种版本:FLUX.1 [pro]、FLUX.1 [dev]和FLUX.1 [schnell]。
  • FLUX.1 [dev]适用于非商业应用,FLUX.1 [schnell]速度更快但质量稍逊。
  • 在MacBook上运行FLUX.1 [dev]时需要解决内存和数据类型问题。
  • 需要在Hugging Face上创建账户并生成访问令牌以下载模型。
  • 设置PYTORCH_MPS_HIGH_WATERMARK_RATIO以解决内存不足的问题。
  • 需要将数据类型从float64更改为float32以兼容MPS框架。
  • FLUX.1 [dev]生成的图像质量较高,能够准确反映提示内容。
  • FLUX.1 [schnell]生成图像速度更快,但质量略低于FLUX.1 [dev]。
  • 实验室计划推出文本到视频系统,进一步拓展生成媒体的能力。
➡️

继续阅读