FLUX.1 Schnell

🚀 FLUX.1 [schnell]

FLUX.1 [schnell] 是一個擁有 120 億參數的校正流變壓器，能夠根據文本描述生成圖像。它在輸出質量和遵循提示方面表現出色，可與閉源模型相媲美。想了解更多信息，請閱讀我們的 博客文章。

🚀 快速開始

我們在專門的 GitHub 倉庫 中提供了 FLUX.1 [schnell] 的參考實現以及採樣代碼。鼓勵希望基於 FLUX.1 [schnell] 進行開發的開發者和創意人士以此為起點。

✨ 主要特性

1. 卓越的輸出質量：輸出質量達到前沿水平，在遵循提示方面表現出色，可與閉源替代方案的性能相匹配。
2. 快速生成圖像：通過潛在對抗擴散蒸餾進行訓練，FLUX.1 [schnell] 僅需 1 到 4 步即可生成高質量圖像。
3. 寬鬆的許可協議：該模型根據 apache-2.0 許可證發佈，可用於個人、科學和商業目的。

💻 使用示例

基礎用法

我們提供了使用 diffusers 庫調用 FLUX.1 [schnell] 的示例代碼。首先，安裝或升級 diffusers 庫：

pip install -U diffusers

然後，你可以使用 FluxPipeline 運行模型：

import torch
from diffusers import FluxPipeline

pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power

prompt = "A cat holding a sign that says hello world"
image = pipe(
    prompt,
    guidance_scale=0.0,
    num_inference_steps=4,
    max_sequence_length=256,
    generator=torch.Generator("cpu").manual_seed(0)
).images[0]
image.save("flux-schnell.png")

更多信息請查看 diffusers 文檔。

API 端點

你還可以通過以下 API 端點使用 FLUX.1 模型：

• bfl.ml（目前為 FLUX.1 [pro]）
• replicate.com
• fal.ai
• mystic.ai

ComfyUI

FLUX.1 [schnell] 也可在 Comfy UI 中使用，支持基於節點的本地推理工作流。

📚 詳細文檔

侷限性

• 此模型並非用於提供事實信息，也不具備該能力。
• 作為統計模型，該檢查點可能會放大現有的社會偏見。
• 模型可能無法生成與提示相匹配的輸出。
• 遵循提示的效果會受到提示風格的嚴重影響。

禁止使用場景

該模型及其衍生產品不得用於以下情況：

• 以任何違反適用的國家、聯邦、州、地方或國際法律或法規的方式使用。
• 用於以任何方式剝削、傷害或試圖剝削、傷害未成年人；包括但不限於招攬、創建、獲取或傳播兒童剝削內容。
• 生成或傳播可證實的虛假信息和/或旨在傷害他人的內容。
• 生成或傳播可用於傷害個人的個人身份信息。
• 騷擾、虐待、威脅、跟蹤或欺凌個人或群體。
• 創建非自願的裸體或非法色情內容。
• 用於對個人合法權利產生不利影響的全自動決策，或以其他方式創建或修改具有約束力的可執行義務。
• 生成或推動大規模的虛假信息宣傳活動。

📄 許可證

本模型採用 apache-2.0 許可證發佈，可用於個人、科學和商業目的。