sd_dreambooth_extension 项目教程

sd_dreambooth_extension 项目教程

sd_dreambooth_extension 项目地址: https://gitcode.com/gh_mirrors/sd/sd_dreambooth_extension

1. 项目介绍

sd_dreambooth_extension 是一个为 Stable-Diffusion-WebUI 开发的扩展项目，旨在优化和增强 DreamBooth 的训练功能。该项目基于 Shivam Shriao 的 Diffusers 仓库，并进行了性能优化，特别适用于低 VRAM 的 GPU。此外，它还整合了 Koyha SS 的部分功能，并添加了多个新特性，如同时训练多个概念和即将推出的修复训练功能。

2. 项目快速启动

安装步骤

1. 打开 Stable-Diffusion WebUI：

   • 启动 Stable-Diffusion WebUI。

2. 安装扩展：

   • 进入 WebUI 的 "Extensions" 标签，选择 "Available" 子标签。
   • 点击 "Load from:" 加载扩展列表。
   • 找到 sd_dreambooth_extension 并点击 "install"。

3. 重启 WebUI：

   • 安装完成后，必须完全重启 Stable-Diffusion WebUI。

4. 设置环境变量（可选）：

   • 为了确保安装顺利，可以设置以下环境变量：

     set/export REQS_FILE=extensions\sd_dreambooth_extension\requirements.txt
     

   • 或者跳过安装过程：

     DREAMBOOTH_SKIP_INSTALL=True
     

启动代码示例

以下是启动 sd_dreambooth_extension 的示例代码：

# 设置环境变量
export REQS_FILE=extensions/sd_dreambooth_extension/requirements.txt
export DREAMBOOTH_SKIP_INSTALL=True

# 启动 Stable-Diffusion WebUI
./webui-user-dreambooth.sh

3. 应用案例和最佳实践

创建模型

1. 进入 DreamBooth 标签：

   • 在 WebUI 中选择 "DreamBooth" 标签。

2. 创建新模型：

   • 在 "Create Model" 子标签中，输入新模型的名称并选择源检查点。
   • 如果使用 HF Hub 的模型，输入模型 URL 和令牌。
   • 点击 "Create" 创建模型。

训练模型

1. 设置训练参数：

   • 在 "Training Parameters" 部分设置训练步数、批量大小等参数。

2. 开始训练：

   • 点击 "Train" 按钮开始训练。

生成样本

1. 生成样本：
   • 在训练过程中，点击 "Generate Samples*" 按钮生成样本。

4. 典型生态项目

Stable-Diffusion-WebUI

Stable-Diffusion-WebUI 是一个基于 Web 的用户界面，用于管理和训练 Stable Diffusion 模型。sd_dreambooth_extension 作为其扩展，提供了更强大的 DreamBooth 训练功能。

Huggingface Diffusers

Huggingface Diffusers 是一个开源库，提供了多种预训练的扩散模型。sd_dreambooth_extension 基于此库进行了优化，特别适用于低 VRAM 的 GPU。

Koyha SS

Koyha SS 是一个用于图像生成的开源项目，sd_dreambooth_extension 整合了其部分功能，进一步增强了图像生成的能力。

通过这些生态项目的结合，sd_dreambooth_extension 能够提供更高效、更灵活的模型训练和图像生成体验。

sd_dreambooth_extension 项目地址: https://gitcode.com/gh_mirrors/sd/sd_dreambooth_extension