前言
本文介绍 FLUX.2 Klein 开源模型的本地部署方案。
一、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. 硬件要求
- 显卡：NVIDIA RTX 3060 8GB/4060 8GB 及以上
- 内存：16GB 及以上
- 硬盘：预留 20GB 空间
- 系统：Windows 10/11 或 Linux
2. 环境安装
# 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
三、部署方案
方式 1：Python 脚本一键运行
步骤 1：创建运行脚本
新建文件 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"
)
pipe.enable_model_cpu_offload()
prompt =
negative_prompt =
image = pipe(
prompt=prompt,
negative_prompt=negative_prompt,
height=,
width=,
guidance_scale=,
num_inference_steps=,
generator=torch.Generator().manual_seed()
).images[]
image.save()
()
Image.().show()
