2.0 KiB
2.0 KiB
使用Dreambooth进行云上模型训练
您可以打开Dreambooth标签页，通过结合使用Dreambooth原生区域及解决方案新增面板Amazon SageMaker Inference，实现调用云上资源的Dreambooth云上模型训练工作。
创建模型
-
打开Dreambooth标签页，模块Model的标签页Create From Cloud。
-
在Name文本框输入所需创建模型名称。
!!! Important "提示" 请注意命名格式要求，只能包含字母数字和“-”。
-
在Source Checkpoint下拉列表，选择一个checkpoint文件。
补充： 此处checkpoint文件包含两个来源的文件：一是以local开头的是本地存储的checkpoint文件，另一是以cloud开头的是存储在S3上的checkpoint文件。首次使用建议选择local checkpoint文件。
4. 点击Create Model From Cloud按钮，开始创建模型。Model Creation Jobs Details区域会即时增加本次模型创建的工作状态。当状态变成Complete，即完成模型创建。
训练模型
- 打开Dreambooth标签页，模块Model的标签页Select From Cloud。
- 刷新Model列表，选择需要训练的模型。
- 在Input模块设置相应的参数。
- 需要选择将模型保存到子目录
- 目前不支持单独保存lora模型，所以请不要勾选下面这个选项
- 需要选择将模型保存到子目录
- 点击SageMaker Train，启动模型训练任务。Training Job Details区域会即时增加本次模型训练的工作状态。当状态变成Complete，同时您部署解决方案时预留的邮箱将会收到邮件通知，即完成模型训练。
- 后续工作。可以进入txt2img标签页的Amazon SageMaker Inference面板，刷新Stable Diffusion Checkpoint，即可看到已训练好的模型。