🚀 彩色画册Flux LoRA模型
本模型是一个用于图像生成的LoRA模型,可根据特定提示词生成彩色画册风格的图像。目前模型仍在训练阶段,后续会不断优化。
🚀 快速开始
环境设置
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 = "prithivMLmods/Coloring-Book-Flux-LoRA"
trigger_word = "Coloring Book"
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
触发词使用
你应该使用 Coloring Book
来触发图像生成。
✨ 主要特性
- 文本到图像生成:根据输入的文本描述生成对应的彩色画册风格图像。
- LoRA模型:基于LoRA技术,可与基础模型结合使用,提升模型的灵活性和效率。
📚 详细文档
模型描述
prithivMLmods/Coloring-Book-Flux-LoRA
图像参数处理如下:
参数 |
值 |
参数 |
值 |
LR调度器 |
constant |
噪声偏移 |
0.03 |
优化器 |
AdamW |
多分辨率噪声折扣 |
0.1 |
网络维度 |
64 |
多分辨率噪声迭代次数 |
10 |
网络Alpha |
32 |
重复与步数 |
20 & 2000 |
轮数 |
10 |
每N轮保存一次 |
1 |
标注方式:florence2 - en(自然语言 & 英语)
训练使用的总图像数:10 [高分辨率]
最佳尺寸
数据源
模型下载
此模型的权重以Safetensors格式提供。
在 [文件与版本] 标签中 下载 它们。
注意事项
⚠️ 重要提示
该模型仍处于训练阶段,这不是最终版本,可能会包含瑕疵,并且在某些情况下表现不佳。
示例展示
以下是一些示例输入和输出:
- 输入文本:'Coloring Book, A black and white drawing of a truck parked in front of a house. The truck is facing towards the right side of the image. There is a large tree to the right of the truck. There are small bushes to the left of the house. A fence is behind the truck on the right. The house has a roof that is made up of wood. The sky above the house is filled with fluffy white clouds.'
输出图像:点击查看
- 输入文本:'Coloring Book, A black and white pencil sketch of a fox standing on its hind legs. The foxs fur is a light brown color, and its ears are a darker brown. Its eyes are black, and the foxs mouth is slightly open, as if it is about to go into the water. Thefoxs ears are sticking up, and it has a black nose and black eyes. There is a tree trunk on the left side of the image, and there are clouds in the sky. There are plants on the right and left of the fox.'
输出图像:点击查看
- 输入文本:'Coloring Book, An eye - level view of a white sports car, the car is facing towards the right. The car is positioned in front of a backdrop of a cityscape, with the city skyline in the background. It is a black and white monochromatic image, with black accents on the cars body and the hood. The front of the car has a large headlight in the center of the front, and a black bumper with a yellow emblem on the right side of the headlight. On the left side, there are two large black tires with silver rims on them.'
输出图像:点击查看
📄 许可证
本模型采用 creativeml - openrail - m 许可证。