Flux Icon Kit LoRA

🚀 Flux-Icon-Kit-LoRA

Flux-Icon-Kit-LoRA 是一個基於 LoRA 技術的文本到圖像生成模型，它以 black-forest-labs/FLUX.1-dev 為基礎模型，能夠根據輸入的文本生成特定風格的圖標圖像。

🚀 快速開始

環境搭建

import torch
from pipelines import DiffusionPipeline

base_model = "black-forest-labs/FLUX.1-dev"
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)

lora_repo = "strangerzonehf/Flux-Icon-Kit-LoRA"
trigger_word = "Icon Kit"  
pipe.load_lora_weights(lora_repo)

device = torch.device("cuda")
pipe.to(device)

觸發詞使用

你需要使用 Icon Kit 來觸發圖像生成。

模型下載

該模型的權重以 Safetensors 格式提供。你可以在 Files & versions 標籤頁 中下載。

✨ 主要特性

• 多樣化圖標生成：能夠根據不同的文本描述生成豐富多樣的圖標圖像，如罐子、南瓜燈、火把等。
• 特定風格呈現：生成的圖標具有獨特的風格，滿足特定的設計需求。

📦 安裝指南

安裝所需的庫和依賴：

import torch
from pipelines import DiffusionPipeline

💻 使用示例

基礎用法

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

# 加載 LoRA 模型
lora_repo = "strangerzonehf/Flux-Icon-Kit-LoRA"
trigger_word = "Icon Kit"  
pipe.load_lora_weights(lora_repo)

# 將模型移動到 GPU 設備
device = torch.device("cuda")
pipe.to(device)

# 輸入文本提示
prompt = "Icon Kit, An animated image of a yellow and purple jar. The jar has a white lightning bolt on it. The background is a bright blue."

# 生成圖像
image = pipe(prompt).images[0]
image.show()

📚 詳細文檔

模型描述

Flux-Icon-Kit-LoRA 是基於 black-forest-labs/FLUX.1-dev 基礎模型的 LoRA 模型，用於生成圖標圖像。

圖像生成參數

標註信息

標註使用 florence2-en（自然語言 & 英語）。

訓練數據

總共使用了 40 張原始 8 位圖像進行訓練。

最佳尺寸與推理

推理範圍

• 推薦推理步數：30 - 35

🔧 技術細節

該模型基於 LoRA 技術對基礎模型進行微調，通過調整一系列參數，如學習率調度器、優化器等，實現對圖標圖像生成的優化。

📄 許可證

本模型使用 creativeml-openrail-m 許可證。