🚀 图像生成模型：Flux-C4C-Design-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-C4C-Design-LoRA"
trigger_word = "Smiley C4C"  
pipe.load_lora_weights(lora_repo)

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

触发词使用

你需要使用Smiley C4C来触发图像生成。

模型下载

该模型的权重以Safetensors格式提供。 点击下载，可在Files & versions选项卡中获取。

✨ 主要特性

• 文本到图像生成：根据输入的文本提示生成对应的图像。
• LoRA微调：基于基础模型进行高效微调，提升特定场景的生成效果。
• 多场景适用：适用于艺术创作、设计等多个领域。

📦 安装指南

安装所需依赖库：

import torch
from pipelines import DiffusionPipeline

💻 使用示例

基础用法

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-C4C-Design-LoRA"
trigger_word = "Smiley C4C"  
pipe.load_lora_weights(lora_repo)

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

# 生成图像示例
prompt = f"{trigger_word}, a close-up of a persons head is visible. The person is wearing a gray sweater, adorned with a silver chain around their neck, and a silver helmet on their head. The helmet is adorned with two cork-like objects, each with a light shining on it, adding a touch of warmth to the scene. The background is a dark blue, creating a stark contrast to the silver helmet."
image = pipe(prompt).images[0]
image.save("output.png")

📚 详细文档

图像生成示例

图像生成参数

最佳尺寸与推理

推理范围

• 推荐推理步数：30–35

📄 许可证

本模型采用creativeml-openrail-m许可证。