Flux Sketch Scribble LoRA

🚀 Flux-Sketch-Scribble-LoRA模型

Flux-Sketch-Scribble-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 = "strangerzonehf/Flux-Sketch-Scribble-LoRA"
trigger_word = "Sketch Scribble"  
pipe.load_lora_weights(lora_repo)

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

觸發詞使用

你需要使用 Sketch Scribble 來觸發圖像生成。

模型下載

該模型的權重以Safetensors格式提供。 點擊下載，可在“Files & versions”標籤中獲取。

✨ 主要特性

• 文本到圖像生成：根據輸入的文本描述，生成對應的黑白素描圖像。
• 多種示例支持：支持人物、動物等不同主題的圖像生成。

📦 安裝指南

在運行代碼前，請確保已經安裝了必要的依賴庫，如torch和diffusers。可以使用以下命令進行安裝：

pip install torch diffusers

💻 使用示例

基礎用法

使用以下代碼生成圖像：

# 假設已經完成上述環境設置
prompt = "Sketch Scribble, A black and white portrait of a mans face. The man has a dark beard and mustache. His eyes are open and he has a serious look on his face. His head is covered in black lines. Behind the mans head is a white background with black lines forming a starburst."
image = pipe(prompt).images[0]
image.save("generated_image.png")

📚 詳細文檔

圖像生成參數

最佳尺寸與推理

推理範圍

• 推薦推理步數：30–35

標註信息

標註採用florence2-en（自然語言 & 英語）。

訓練圖像總數

總共使用了20張圖像進行訓練。

📄 許可證

本模型遵循Apache 2.0許可證。