🚀 simpletuner-finetuned-sd3
这是一个基于 sd3/unknown-model 的全秩微调模型。该模型主要用于文本到图像的生成任务,能够根据输入的文本生成高质量的图像。
🚀 快速开始
此模型是基于 sd3/unknown-model 的全秩微调模型。训练期间使用的主要验证提示为:
A photo-realistic pathology image
✨ 主要特性
- 文本到图像生成:支持根据输入的文本生成相应的图像。
- 图像到图像转换:可对输入的图像进行转换。
- 多种参数设置:提供丰富的训练和验证参数设置。
📦 安装指南
文档未提及安装步骤,暂不提供。
💻 使用示例
基础用法
import torch
from diffusers import DiffusionPipeline
model_id = 'Minh-Ha/simpletuner-finetuned-sd3'
pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
prompt = "A photo-realistic pathology image"
negative_prompt = 'blurry, cropped, ugly'
pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
model_output = pipeline(
prompt=prompt,
negative_prompt=negative_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(42),
width=1024,
height=1024,
guidance_scale=3.0,
).images[0]
model_output.save("output.png", format="PNG")
📚 详细文档
验证设置
- CFG:
3.0
- CFG Rescale:
0.0
- 步数:
20
- 采样器:
FlowMatchEulerDiscreteScheduler
- 种子:
42
- 分辨率:
1024x1024
- 跳过层引导:
注意:验证设置不一定与训练设置相同。
你可以在以下图库中找到一些示例图像:
文本编码器未进行训练。你可以重用基础模型的文本编码器进行推理。
训练设置
- 训练轮数: 1
- 训练步数: 5000
- 学习率: 5e-06
- 最大梯度值: 2.0
- 有效批量大小: 4
- 微批量大小: 1
- 梯度累积步数: 4
- GPU 数量: 1
- 梯度检查点: True
- 预测类型: flow_matching (额外参数=['shift=3'])
- 优化器: adamw_bf16
- 可训练参数精度: Pure BF16
- 基础模型精度:
no_change
- 字幕丢弃概率: 0.1%
数据集
images-512
- 重复次数: 1
- 图像总数: 3061
- 纵横比桶总数: 1
- 分辨率: 0.262144 兆像素
- 裁剪: True
- 裁剪样式: 居中
- 裁剪纵横比: 正方形
- 用于正则化数据: 否
images-768
- 重复次数: 1
- 图像总数: 2242
- 纵横比桶总数: 1
- 分辨率: 0.589824 兆像素
- 裁剪: True
- 裁剪样式: 居中
- 裁剪纵横比: 正方形
- 用于正则化数据: 否
images-1024
- 重复次数: 1
- 图像总数: 1449
- 纵横比桶总数: 1
- 分辨率: 1.048576 兆像素
- 裁剪: True
- 裁剪样式: 居中
- 裁剪纵横比: 正方形
- 用于正则化数据: 否
🔧 技术细节
文档未提供足够的技术细节描述,暂不展示。
📄 许可证
本项目使用其他许可证。