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，你可以通過 此鏈接 查看許可證詳情。

📦 模型信息

🔍 示例展示