在 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]。
- 实验室计划推出文本到视频系统,进一步拓展生成媒体的能力。
➡️