FLUX.1 Dev LoRA HuggingFaceIcons V0.1

✨ 主要特性

• 增強了 FLUX.1 [dev] 生成 Hugging Face 表情符號圖標的能力，可用於在 Hugging Face 上開展新項目。
• 使用觸發詞 HuggingFaceIcons 激活特定的 Hugging Face 圖標生成功能。

📦 安裝指南

首先，確保你已經安裝了 FLUX.1 [dev]：

# 安裝
pip install -U diffusers

# 使用 Diffusers 加載
import torch
from diffusers import FluxPipeline

# 加載基礎模型
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload()

# 加載並應用 LoRA 權重
pipe.load_lora_weights("Borcherding/FLUX.1-dev-LoRA-HuggingFaceIcons-v0.1")

# 生成圖像
prompt = "HuggingFaceIcons a yellow smiley face sun with a book in its hands, surrounded by planets and stars in a dark background."
image = pipe(
    prompt,
    height=1024,
    width=1024,
    guidance_scale=3.5,
    num_inference_steps=50,
    max_sequence_length=512,
    generator=torch.Generator("cpu").manual_seed(0)
).images[0]
image.save("autumn-scene.png")

# 如有需要，卸載 LoRA 權重
pipe.unload_lora_weights()

# 可選：永久合併 LoRA 權重
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.load_lora_weights("Borcherding/FLUX.1-dev-LoRA-HuggingFaceIcons-v0.1")
pipe.merge_lora_weights()
pipe.save_pretrained("FLUX.1-dev-LoRA-HuggingFaceIcons-v0.1")

💻 使用示例

基礎用法

你可以使用以下示例代碼進行圖像生成：

# 加載基礎模型
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload()

# 加載並應用 LoRA 權重
pipe.load_lora_weights("Borcherding/FLUX.1-dev-LoRA-HuggingFaceIcons-v0.1")

# 生成圖像
prompt = "HuggingFaceIcons a yellow smiley face sun with a book in its hands, surrounded by planets and stars in a dark background."
image = pipe(
    prompt,
    height=1024,
    width=1024,
    guidance_scale=3.5,
    num_inference_steps=50,
    max_sequence_length=512,
    generator=torch.Generator("cpu").manual_seed(0)
).images[0]
image.save("autumn-scene.png")

高級用法

如果你需要永久合併 LoRA 權重，可以使用以下代碼：

# 加載基礎模型
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload()

# 加載並應用 LoRA 權重
pipe.load_lora_weights("Borcherding/FLUX.1-dev-LoRA-HuggingFaceIcons-v0.1")

# 合併 LoRA 權重
pipe.merge_lora_weights()

# 保存模型
pipe.save_pretrained("FLUX.1-dev-LoRA-HuggingFaceIcons-v0.1")

📚 詳細文檔

觸發詞

你應該使用 HuggingFaceIcons 來觸發圖像生成。將觸發詞放在提示詞的開頭可獲得最佳效果。

提示詞示例

• "HuggingFaceIcons a yellow smiley face emoji with its hands on its face, set against a black background. It is an animated emoji, with a cheerful expression and bright colors."
• "HuggingFaceIcons a yellow smiley face emoji watering a potted plant with a blue watering can. The plant is in a pot and the background is white."
• "HuggingFaceIcons a yellow smiley face sun with a book in its hands, surrounded by planets and stars in a dark background."

特殊功能

可以生成黃色笑臉表情符號，手持物品、穿著噴氣揹包等特定樣式的圖標。

模型下載

該模型的權重以 Safetensors 格式提供。 點擊下載，可在“文件與版本”選項卡中獲取。

🔧 技術細節

• 數據集：使用了 Borcherding/HuggingFaceIcons-imageAnnotations-v0.1 數據集。
• 基礎模型：black-forest-labs/FLUX.1-dev。
• 訓練方式：使用 Fluxgym 進行訓練，詳情見：https://github.com/cocktailpeanut/fluxgym 。

📄 許可證

本 LoRA 遵循與 FLUX.1 [dev] 相同的許可條款。請參考基礎模型的許可證以瞭解使用條款。

版權信息

• 基礎模型：由 Black Forest Labs 開發的 FLUX.1 [dev]。
• LoRA 訓練與開發：[Borcherding at BorchInk]

侷限性

• 這是 FLUX.1 [dev] 的 LoRA，需要基礎模型才能運行。
• 專注於生成 Hugging Face 圖標。
• 在小數據集上進行訓練。
• 繼承了 FLUX.1 [dev] 的基礎侷限性。