许可证：其他
许可证名称：flux-1-dev-non-commercial-license
许可证链接：https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/LICENSE.md
基础模型：

• black-forest-labs/FLUX.1-dev
  任务标签：文本生成图像
  库名称：diffusers
  标签：
• flux
• 文本生成图像

Flux.1 Lite

我们激动地宣布推出Flux.1 Lite，这是一个从FLUX.1-dev模型蒸馏而来的80亿参数Transformer模型。该版本减少了7GB内存占用，运行速度提升了23%，同时保持了与原始模型相同的精度（bfloat16）。

🔥 更新 🔥：我们发布了Flux.1 Lite 8B的新版本。该版本使用新数据集训练，效果优于之前的Alpha版本。主要改进包括：

• 支持更广的引导值范围（2.0-5.0）蒸馏
• 支持更广的步数范围（20-32）蒸馏
• 包含更长提示词的多样化数据集

文本生成图像

Flux.1 Lite已准备好释放您的创造力！为获得最佳效果，我们强烈推荐使用2.0至5.0的guidance_scale，并将n_steps设置在20至32之间。

import torch  
from diffusers import FluxPipeline  

torch_dtype = torch.bfloat16  
device = "cuda"  

# 加载模型  
model_id = "Freepik/flux.1-lite-8B"  
pipe = FluxPipeline.from_pretrained(  
    model_id, torch_dtype=torch_dtype  
).to(device)  

# 推理  
prompt = "特写镜头：一只荧光皮肤的绿色外星人站在深紫色森林中央"  

guidance_scale = 3.5  
n_steps = 28  
seed = 11  

with torch.inference_mode():  
    image = pipe(  
        prompt=prompt,  
        generator=torch.Generator(device="cpu").manual_seed(seed),  
        num_inference_steps=n_steps,  
        guidance_scale=guidance_scale,  
        height=1024,  
        width=1024,  
    ).images[0]  
image.save("output.png")  

研发动机

受Ostris研究的启发，我们通过分析各模块输入输出的均方误差（MSE）量化其对最终结果的贡献，发现存在显著差异。

正如Ostris指出的，并非所有模块贡献均等。跳过早期MMDiT或晚期DiT的单个模块会显著影响模型表现，而跳过中间任何单个模块对最终图像质量影响不大。

ComfyUI集成

我们还设计了ComfyUI工作流，让Flux.1 Lite使用更便捷！文件见comfy/flux.1-lite_workflow.json。

模型检查点下载：flux.1-lite-8B.safetensors

在Freepik体验！

我们的AI生成器现已搭载Flux.1 Lite！

🔥 动态 🔥

• 2024年12月30日：Flux.1 Lite 8B新版模型发布于HuggingFace仓库
• 2024年10月23日：Alpha 8B检查点发布于HuggingFace仓库

引用

如果您觉得我们的工作有帮助，请引用！

@article{flux1-lite,  
  title={Flux.1 Lite：高效文本生成图像的FLUX1.dev蒸馏模型},  
  author={Daniel Verdú, Javier Martín},  
  email={dverdu@freepik.com, javier.martin@freepik.com},  
  year={2024},  
}