关于这个LoRA

这是为FLUX.1-dev文生图模型训练的LoRA适配器，支持通过diffusers或ComfyUI调用。

模型训练使用Replicate平台的AI工具包完成：https://replicate.com/ostris/flux-dev-lora-trainer/train

触发词

使用嗨呀作为生成图像的触发词。

通过Replicate API调用

import replicate

input = {
    "prompt": "嗨呀",
    "lora_weights": "https://huggingface.co/guokee/hiya/resolve/main/lora.safetensors"
}

output = replicate.run(
    "black-forest-labs/flux-dev-lora",
    input=input
)
for index, item in enumerate(output):
    with open(f"output_{index}.webp", "wb") as file:
        file.write(item.read())

使用🧨 diffusers库加载

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
pipeline.load_lora_weights('guokee/hiya', weight_name='lora.safetensors')
image = pipeline('嗨呀').images[0]

更多技术细节（包括权重调整、LoRA合并与融合）请参阅diffusers官方文档

训练参数

• 训练步数：1000
• 学习率：0.0004
• LoRA秩：16

贡献你的作品

欢迎在社区讨论区分享你用这个LoRA创作的作品。