Sweet Brush

🚀 sweet-brush

sweet-brush是一个文本到图像的模型，借助AI Toolkit by Ostris进行训练，能生成具有特定风格的图像，可用于多种图像生成工具。

🚀 快速开始

触发词

你需要使用 in the style of TOK 来触发图像生成。

下载模型并在不同工具中使用

此模型的权重以Safetensors格式提供。你可以在 Files & versions 标签页 下载。下载后可在ComfyUI、AUTOMATIC1111、SD.Next、Invoke AI等工具中使用。

使用 🧨 diffusers 库

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('fffiloni/sweet-brush', weight_name='sweet-brush.safetensors')
image = pipeline('A person in a bustling cafe. in the style of TOK').images[0]
image.save("my_image.png")

有关更多详细信息，包括LoRA的加权、合并和融合，请查看 diffusers中加载LoRA的文档。

💻 使用示例

基础用法

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('fffiloni/sweet-brush', weight_name='sweet-brush.safetensors')
image = pipeline('A person in a bustling cafe. in the style of TOK').images[0]
image.save("my_image.png")

高级用法

目前文档未提供高级用法示例，你可以参考 diffusers中加载LoRA的文档 进行更复杂的操作。

📄 许可证

本模型使用的许可证为 flux-1-dev-non-commercial-license，你可以通过 此链接 查看许可证详情。

📦 模型信息

🔍 示例展示