🚀 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 许可证。