文本编码器未参与训练。
推理时可复用基础模型的文本编码器。

训练参数

• 训练周期：142
• 训练步数：5000
• 学习率：0.0001
• 有效批次大小：1
  • 微批次大小：1
  • 梯度累积步数：1
  • GPU数量：1
• 预测类型：flow-matching
• 重缩放零SNR beta值：否
• 优化器：adamw_bf16
• 精度：bf16
• 量化：是（int8-quanto）
• Xformers：未使用
• LoRA秩：4
• LoRA Alpha：无
• LoRA丢弃率：0.1
• LoRA初始化风格：默认

数据集

动漫测试01

• 重复次数：0
• 总图片数：35
• 总宽高比分组：1
• 分辨率：0.262144百万像素
• 裁剪：是
• 裁剪风格：居中
• 裁剪比例：正方形

推理

import torch
from diffusers import DiffusionPipeline

model_id = 'black-forest-labs/FLUX.1-dev'
adapter_id = 'Disra/lora-anime-test-02'
pipeline = DiffusionPipeline.from_pretrained(model_id)
pipeline.load_lora_weights(adapter_id)

prompt = "动漫风格数字艺术作品，描绘一位蓝绿发色、绿眼睛、穿着连体泳衣的女孩"

pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
image = pipeline(
    prompt=prompt,
    num_inference_steps=20,
    generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
    width=1024,
    height=1024,
    guidance_scale=3.5,
).images[0]
image.save("output.png", format="PNG")