在 MacBook (M2) 上运行由 Stable Diffusion 原始开发者推出的 FLUX.1 图像生成 AI 模型 ([dev]/[schnell])
内容提要
黑森林实验室推出了开源图像生成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]。
-
实验室计划推出文本到视频系统,进一步拓展生成媒体的能力。
延伸问答
FLUX.1模型有哪些版本?
FLUX.1模型有三个版本:FLUX.1 [pro]、FLUX.1 [dev]和FLUX.1 [schnell]。
如何在MacBook上运行FLUX.1 [dev]模型?
在MacBook上运行FLUX.1 [dev]模型需要解决内存和数据类型问题,并在Hugging Face上创建账户以下载模型。
FLUX.1 [schnell]与FLUX.1 [dev]有什么区别?
FLUX.1 [schnell]速度更快,但生成的图像质量略低于FLUX.1 [dev]。
在运行FLUX.1时如何解决内存不足的问题?
可以通过设置PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0来解决内存不足的问题,但需谨慎以免系统崩溃。
如何创建Hugging Face的访问令牌?
在Hugging Face上创建账户后,进入设置中的访问令牌部分,选择创建新令牌并生成。
FLUX.1模型的未来发展计划是什么?
实验室计划推出文本到视频系统,以进一步拓展生成媒体的能力。