高性能应用服务
有奖捉虫：行业应用 & 管理与支持文档专题 HOT
文档中心 > 高性能应用服务 > 最佳实践 > 快速构建 Stable Diffusion 文生图 API 服务

快速构建 Stable Diffusion 文生图 API 服务

最近更新时间：2024-04-15 15:56:41

我的收藏

本页目录：

本次我们使用 腾讯云高性能应用服务 HAI 体验快速搭建并使用 AI 模型 StableDiffusion ，实现思路如下:
提前通过高性能应用服务 HAI 部署成功 StableDiffusion 应用。
基于部署好的应用，利用体验 JupyterLab 进行 StableDiffusion API 的部署。

前提

在部署 API 服务之前，请确保您已成功部署 StableDiffusion 应用。详细步骤可参见 快速使用 Stable Diffusion 文生图应用。

部署 API 服务

1. 进入 jupyter_lab 控制台操作界面。
1.1 在实例列表中选择更多 > JupyterLab 并进入该实例的详情页。



1.2 初步认识并操作 JupyterLab。

1.3 选择使用终端命令行操作。

输入代码：
cd stable-diffusion-webui
python launch.py --nowebui --xformers --opt-split-attention  --listen --port 7862

命令参数描述如下图：
命令
描述
--nowebui
以 API 模式启动。
--xformers
改善内存消耗和速度。
--opt-split-attention
Cross attention layer optimization 优化显着减少了内存使用。
--listen
默认启动绑定的 IP 是 127.0.0.1。
--port
默认端口是7860，可以配置并修改该参数，例如：--port 7862。
--gradio-auth username:password
如果希望给 WebUI 设置登录密码，可以配置该参数，例如：--gradio-auth GitLqr:123456。
操作截图如下图所示：

1.4 添加高性能应用服务 HAI 的端口配置，使外部网络能够顺利地访问该服务器提供的 API 服务。
1.4.1 在算力管理页面。单击实例空白进入详情设置页。
1.4.2 在端口配置弹窗中，单击编辑规则。

1.4.3 在安全组规则页面中，在入站规则页签单击添加规则。

配置参考如下：
来源：0.0.0.0/0 
协议端口：TCP:7862 (根据您配置的端口填写)

2. 启动 StableDiffusion API 接口使用指南
2.1 配置完成后，在浏览器地址栏输入服务器 IP 地址:端口号/docs 可查看相关的 API 接口使用指南。
官方提供的常用 API 如下：
/sdapi/v1/txt2img文字生图 POST
/sdapi/v1/img2img图片生图 POST
/sdapi/v1/options获取设置 GET | 更新设置 POST（可用来更新远端的模型）
/sdapi/v1/sd-models获取所有的模型 GET

2.2 查看相关接口示例 (/sdapi/v1/txt2img ) 。
常用输入如下：
{
     "denoising_strength": 0,
     "prompt": "puppy dogs",
     "negative_prompt": "",
     "seed": -1,
     "batch_size": 2,
     "n_iter": 1,
     "steps": 50,
     "cfg_scale": 7,
     "width": 512,
     "height": 512,
     "restore_faces": false,
     "tiling": false,
     "sampler_index": "Euler" 
}
可复制以上参数到 Request body 中。
名称
说明
prompt
提示词
negative_prompt
反向提示词
seed
种子，随机数
batch_size
每次张数
n_iter
生成批次
steps
生成步数
cfg_scale
关键词相关性
width
宽度
height
高度
restore_faces
脸部修复
tiling
可平铺
sampler_index
采样方法



请求 API 接口成功截图如下：



返回的格式如下：
{
   "images": [...]，// 这里是一个base64格式的字符串数组，根据请求的图片数量而定
   "parameters": {  ... },//此处为输入的body 
   "info": "{...}"// 返回的图片的信息
}
当看到类似上图的消息时，说明已经成功与远端的服务器进行连接！如果希望验证结果的图片的实际展示效果，可以复制 images 中的其中一张图片的 base64 格式的字符串，到相关的网站下转换为 jpg 格式。
3. 使用 Python 向高性能应用服务 HAI 提供的 StableDiffusionAPI 发送请求。
以下演示如何使用 Python 向 StableDiffusion API 发出请求。向应用程序的 txt2img（即文本到图像）API 发送 POST 请求以简单地生成图像。
我们将使用 requests 包，如果您还没有安装，请使用安装脚本：
pip install requests
我们可以发送一个包含提示的请求作为一个简单的字符串。服务器将返回一个图像作为 base64 编码的 PNG 文件，我们需要对其进行解码。 要解码 base64 图像，我们只需使用 base64.b64decode(b64_image)。 以下使用 Python 作为脚本代码测试：
import json
import base64
import requests
 
your_ip = '0.0.0.0' # HAI服务器IP地址
your_port =7862  # SD api 监听的端口
 
def submit_post(url: str,data: dict):
  """
  Submit a POST request to the given URL withthe given data.
  """
  return requests.post(url,data=json.dumps(data))
 
def save_encoded_image(b64_image: str,output_path: str):
  """
  Save the given image to the given outputpath.
  """
  with open(output_path,"wb") asimage_file:
      image_file.write(base64.b64decode(b64_image))
 
if __name__ == '__main__':
  #/sdapi/v1/txt2img
  txt2img_url = f'http://{your_ip}:{your_port}/sdapi/v1/txt2img'
  data = {
     'prompt': 'a pretty cat,cyberpunk art,kerem beyit,verycute robot zen,Playful,Independent,beeple |',
     'negative_prompt':'(deformed,distorted,disfigured:1.0),poorlydrawn,bad anatomy,wrong anatomy,extra limb,missing limb,floating limbs,(mutatedhands and fingers:1.5),disconnectedlimbs,mutation,mutated,ugly,disgusting,blurry,amputation,flowers,human,man,woman',
     'Steps':50,
     'Seed':1791574510
  }
  response = submit_post(txt2img_url,data)
  save_encoded_image(response.json()['images'][0],'cat.png')
请记住，您的结果会与上述示例有所不同。 如果遇到问题，请仔细检查运行 StableDiffusionAPI 应用程序的终端的输出。 如果您遇到404 Not Found 的问题，请仔细检查 URL 是否输入正确并指向正确的地址（例如 127.0.0.1）。



服务端可查看每一次接口调用详情：


中国站