PythonAI算法
FLUX.2 Klein 开源模型本地部署与使用指南
FLUX.2 Klein 是 Black Forest Labs 发布的轻量级 AI 绘画模型。该模型的硬件要求、环境配置以及两种本地部署方式（Python 脚本与 ComfyUI）。内容涵盖模型下载、显存优化、常见问题解决等实操步骤，旨在帮助开发者在消费级显卡上高效运行 AI 绘画任务，支持文生图、图生图及图像编辑功能。
FLUX.2 Klein 是 Black Forest Labs 发布的轻量级 AI 绘画模型。该模型的硬件要求、环境配置以及两种本地部署方式（Python 脚本与 ComfyUI）。内容涵盖模型下载、显存优化、常见问题解决等实操步骤，旨在帮助开发者在消费级显卡上高效运行 AI 绘画任务，支持文生图、图生图及图像编辑功能。
本文介绍 FLUX.2 Klein 开源模型的本地部署方案。
FLUX.2 Klein 是黑森林实验室（Black Forest Labs）最新发布的紧凑型 AI 绘画模型，主打亚秒级生成、低显存占用及全能编辑能力。
| 对比项 | FLUX.2 Klein（4B 版） | FLUX.1 | SDXL 1.0 |
|---|---|---|---|
| 参数规模 | 4B/9B（轻量化） | 12B | 3.5B |
| 显存要求 | 8GB+（RTX4060 可跑） | 16GB+ | 10GB+ |
| 生成速度 | 0.5-1 秒/张（1024×1024） | 2-3 秒/张 | 5-8 秒/张 |
| 核心能力 | 文生图 + 图生图 + 图像编辑 | 仅文生图 | 文生图 + 图生图 |
| 授权协议 | Apache-2.0（商用友好） | 非商用 | Apache-2.0 |
简单说：4B 版 8G 显存就能跑，速度快，支持图像编辑，商用友好。
# 1. 安装核心依赖
pip install -U diffusers transformers torch accelerate safetensors
# 2. 安装图像工具
pip install pillow matplotlib
# 3. 安装 huggingface 工具
pip install huggingface_hub
提示：如果下载慢，可换国内镜像源，在命令后加
-i https://pypi.tuna.tsinghua.edu.cn/simple
新建文件 flux2_klein_demo.py，代码如下：
import torch
from diffusers import Flux2KleinPipeline
from PIL import Image
# 1. 加载模型
model_id = "black-forest-labs/FLUX.2-klein-4B"
dtype = torch.bfloat16
pipe = Flux2KleinPipeline.from_pretrained(
model_id,
torch_dtype=dtype,
cache_dir="./models"
)
# 2. 显存优化
pipe.enable_model_cpu_offload()
# pipe.enable_vae_slicing()
# 3. 生成图片
prompt = "一只戴着墨镜的橘猫，坐在复古摩托车上，赛博朋克风格，高清细节，1024×1024"
negative_prompt = "模糊，低分辨率，畸形，多余肢体，水印"
image = pipe(
prompt=prompt,
negative_prompt=negative_prompt,
height=1024,
width=1024,
guidance_scale=4.0,
num_inference_steps=20,
generator=torch.Generator("cuda").manual_seed(42)
).images[0]
# 4. 保存并查看图片
image.save("flux2_klein_cat.png")
print("图片生成完成！已保存为 flux2_klein_cat.png")
Image.open("flux2_klein_cat.png").show()
在脚本所在文件夹打开终端，执行：
python flux2_klein_demo.py
首次运行会自动下载模型，后续运行直接生成图片。
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI
pip install -r requirements.txt
将以下模型文件放入对应目录：
ComfyUI/models/clip/
ComfyUI/models/unet/
ComfyUI/models/vae/
python main.py
访问 http://127.0.0.1:8188，选择工作流即可生成。
pipe.enable_model_cpu_offload()HF_ENDPOINT=https://hf-mirror.com./models 目录guidance_scale（4-6 最佳）FLUX.2 Klein 降低了 AI 绘画的本地部署门槛，8G 显存即可实现快速生成，支持图像编辑，适合个人创作及小型商用。
微信公众号「极客日志」，在微信中扫描左侧二维码关注。展示文案：极客日志 zeeklog
使用加密算法（如AES、TripleDES、Rabbit或RC4）加密和解密文本明文。 在线工具，加密/解密文本在线工具，online
生成新的随机RSA私钥和公钥pem证书。 在线工具，RSA密钥对生成器在线工具，online
基于 Mermaid.js 实时预览流程图、时序图等图表，支持源码编辑与即时渲染。 在线工具，Mermaid 预览与可视化编辑在线工具，online
解析常见 curl 参数并生成 fetch、axios、PHP curl 或 Python requests 示例代码。 在线工具，curl 转代码在线工具，online
将字符串编码和解码为其 Base64 格式表示形式即可。 在线工具，Base64 字符串编码/解码在线工具，online
将字符串、文件或图像转换为其 Base64 表示形式。 在线工具，Base64 文件转换器在线工具，online