部署及微调Stable Diffusion V1.5模型实现文生图

本文为您介绍如何部署Stable Diffusion V1.5模型以及启动WebUI应用来使用和调试模型。

前提条件

已创建OSS Bucket存储空间,具体操作请参见控制台创建存储空间

一、进入模型详情页面

  1. 进入Model Gallery页面。

    1. 登录PAI控制台

    2. 在左侧导航栏单击工作空间列表,在工作空间列表页面中单击待操作的工作空间名称,进入对应工作空间内。

    3. 在左侧导航栏单击快速开始 > Model Gallery,进入Model Gallery页面。

  2. Model Gallery页面,搜索并单击Stable_Diffusion_V1.5模型卡片,进入模型详情页面。

二、直接部署和调试模型

  1. 在模型详情页面,单击页面右上角的部署按钮,然后在部署配置面板,选择部署资源(例如ecs.gn6e-c12g1.3xlarge),其他参数使用默认配置即可。

  2. 单击部署,并在计费提醒对话框,单击确定

    页面将自动跳转到服务详情页面,您可以在基本信息区域查看模型服务的部署状态。当状态变为运行中时,表明模型服务已成功部署。

  3. 在服务详情页面右上角,单击查看WEB应用,即可启动WebUI应用页面。

  4. WebUI应用页面中,测试模型效果。

    文生图页签的提示词区域中,输入文本内容:The eagle is flying in the sky, in the distance is the vast snow, and under the snow mountain is a grassland,然后单击生成,即可输出下图的推理结果。image..png

三、微调训练模型

您可以对Stable Diffusion V1.5模型进行微调训练,让其更符合实际的业务需求。

  1. 返回到模型详情页面,单击右上角训练按钮。

  2. 训练配置面板中,依次展开各种训练配置选项,查看或修改以下关键配置。其他参数取默认值即可。更多参数配置详情,请参见训练模型

    配置选项

    描述

    数据集配置

    PAI准备了默认数据集,本方案使用默认数据集。您也可以按照以下操作步骤准备自己的数据集。

    1. 在模型详情页面,参照训练数据格式来准备训练数据。您可以参考示例数据集image.png

    2. 将训练数据上传到OSS存储空间,如何上传数据到OSS存储空间,详情请参见控制台上传文件

    3. 更新训练数据集,具体操作,请参见训练模型

    超参数配置

    training_method使用默认配置:standard,其他参数保持不变。

    输出配置

    配置模型输出路径OSS Bucket路径。如何创建OSS Bucket,请参见控制台快速入门

  3. 单击训练

    页面自动跳转到任务详情页面。

训练任务运行成功后,您可以在任务详情页面右上角,单击部署按钮来部署模型。模型部署流程与直接部署模型一致,详情请参见二、直接部署模型和调试模型