文档

面向AIGC文生图模型的Prompt自动化生成

更新时间:

本文为您介绍如何在EAS上部署AIGC Stable Diffusion SDWebUI绘画应用,同时加载PAI自研的BeautifulPrompt插件,以实现Prompt的自动化生成功能,并使用对应的Prompt进行模型推理。

背景信息

AIGC,即通过人工智能技术自动生成内容的生产方式,已经成为继互联网时代的下一个产业时代风口。其中,Stable Diffusion文生图模型是一种开源且流行的跨模态生成模型,旨在生成与给定文本相对应的图像。在Stable Diffusion中,编写Prompt是生成高质量图像的关键。BeautifulPrompt插件利用定制化的大语言模型(LLM)自动实现Prompt的扩写和美化。使用Stable Diffusion文生图模型作为底座,使用和不使用BeautifulPrompt插件效果对比如下。

输入Prompt

不使用BeautifulPrompt的效果

使用BeautifulPrompt的效果

a cat

image.png

image.png

a cute girl

image.png

image.png

a giant tiger

image.png

image.png

前提条件

部署EAS服务

将Stable Diffusion模型部署为AI-Web应用服务,具体操作步骤如下:

  1. 进入PAI-EAS 模型在线服务页面。

    1. 登录PAI控制台

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

    3. 在工作空间页面的左侧导航栏选择模型部署>模型在线服务(EAS),进入PAI-EAS 模型在线服务页面。image.png

  2. PAI-EAS 模型在线服务页面,单击部署服务,在弹出对话框中,选择自定义部署,然后单击确定

  3. 部署服务页面,配置以下关键参数。

    参数

    描述

    服务名称

    自定义服务名称。本案例使用的示例值为:sdwebui_prompt

    部署方式

    选择镜像部署AI-Web应用

    镜像选择

    PAI平台镜像列表中选择stable-diffusion-webui;镜像版本选择4.0-standard

    说明

    由于版本迭代迅速,部署时镜像版本选择最高版本即可。

    运行命令

    系统会自动配置运行命令,您需要在运行命令中添加--public-cache。示例如下:./webui.sh --listen --port 8000 --skip-version-check --no-hashing --no-download-sd-model --skip-install --api --api-log --time-log --sd-dynamic-cache --public-cache

    资源组种类

    选择公共资源组

    资源配置方法

    选择常规资源配置

    资源配置选择

    必须选择GPU类型,实例规格推荐使用ml.gu7i.c16m60.1-gu30(性价比最高)。

    49a4ad71dacfc06efadcbcf6773dd939.png

  4. 单击部署,等待一段时间即可完成模型部署。

    模型状态运行中时,服务部署成功。

启动WebUI进行模型推理

  1. 启动WebUI。

    在服务列表中,单击目标服务的服务方式列下的查看Web应用,来启动WebUI。image.png

  2. 加载BeautifulPrompt插件。

    扩展页签下,检查是否已选中BeautifulPrompt。如果没有选中,您需要选中BeautifulPrompt,并单击应用更改并重载前端来重新加载BeautifulPrompt插件。image.png

  3. 生成Prompt进行模型推理。

    1. 切换到BeautifulPrompt页签,在文本框中输入一个比较简单的Prompt,然后单击生成,即可生成更详细的Prompt。image.png

      PAI提供了多个Prompt生成模型,每个模型生成的Prompt略有不同。其中:

      • pai-bloom-1b1-text2prompt-sd-v2:擅长生成复杂场景下的Prompt。

      • pai-bloom-1b1-text2prompt-sd:生成描述单个物体的Prompt。

      您可以根据场景需要选择合适的模型来生成Prompt。image.png

    2. 选择需要使用的Prompt,单击Prompt右侧的to txt2img

      页面自动跳转至文生图页签,并自动填充了提示词(Prompt)区域。单击生成,即可开启您的AIGC之旅。image.png

      与不使用BeautifulPrompt插件相比,使用BeautifulPrompt插件可以提升生成图像的美观度,并且帮助您补充更多细节。相反,如果不使用BeautifulPrompt插件,则效果如下。image.png

  • 本页导读 (1)
文档反馈