Icon Generator

🚀 圖標生成器

這是一個基於 black-forest-labs/FLUX.1-dev 的標準PEFT LoRA模型，可用於生成簡約風格的圖標。

🚀 快速開始

本項目是一個圖標生成器，基於特定的基礎模型派生而來，可用於生成各種簡約風格的圖標。以下是使用該生成器的基本步驟和相關信息。

✨ 主要特性

• 基於 black-forest-labs/FLUX.1-dev 基礎模型派生，具有良好的生成能力。
• 可生成多種簡約風格的圖標，如箭頭、笑臉等。
• 提供了詳細的訓練和驗證設置，便於復現和調整。

📦 安裝指南

此文檔未提及具體安裝命令，故跳過此章節。

💻 使用示例

基礎用法

import torch
from diffusers import DiffusionPipeline

model_id = 'black-forest-labs/FLUX.1-dev'
adapter_id = 'noahyoungs/icon-generator'
pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16) # loading directly in bf16
pipeline.load_lora_weights(adapter_id)

prompt = "Minimalist icon, arrow up"


## Optional: quantise the model to save on vram.
## Note: The model was quantised during training, and so it is recommended to do the same during inference time.
from optimum.quanto import quantize, freeze, qint8
quantize(pipeline.transformer, weights=qint8)
freeze(pipeline.transformer)
    
pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu') # the pipeline is already in its target precision level
image = pipeline(
    prompt=prompt,
    num_inference_steps=20,
    generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(42),
    width=1024,
    height=1024,
    guidance_scale=3.0,
).images[0]
image.save("output.png", format="PNG")

📚 詳細文檔

驗證設置

• CFG：3.0
• CFG Rescale：0.0
• 步數：20
• 採樣器：FlowMatchEulerDiscreteScheduler
• 種子：42
• 分辨率：1024x1024
• 跳過層引導：無

注意：驗證設置不一定與訓練設置相同。

你可以在以下圖庫中找到一些示例圖像：

文本編碼器未進行訓練，你可以複用基礎模型的文本編碼器進行推理。

訓練設置

數據集

tabler-icons-1024

• 重複次數：10
• 圖像總數：4739
• 縱橫比桶總數：1
• 分辨率：1.048576兆像素
• 裁剪：否
• 裁剪風格：無
• 裁剪縱橫比：無
• 用於正則化數據：否

推理

推理代碼示例見使用示例。

🔧 技術細節

本項目基於 black-forest-labs/FLUX.1-dev 基礎模型，通過PEFT LoRA技術進行微調。在訓練過程中，使用了特定的數據集和訓練設置，以生成簡約風格的圖標。推理時，加載基礎模型和LoRA權重，並根據提示生成圖標。

📄 許可證

許可證類型：其他