如何基于Knative打造生产可用的Stable Diffusion服务_容器服务 Kubernetes 版 ACK(ACK)-阿里云帮助中心

您可以通过Knative解决方案在AIGC领域实现按需使用的Serverless能力，并且做到并发请求精准处理，实现自动弹性扩缩容。本文介绍如何基于Knative打造生产可用的Stable Diffusion服务。

索引

前提条件
背景信息
步骤一：部署Stable Diffusion服务
步骤二：访问服务
步骤三：基于请求实现自动弹性扩缩容
步骤四：查看Stable Diffusion服务监控数据

前提条件

背景信息

重要

阿里云不对第三方模型“Stable Diffusion”的合法性、安全性、准确性进行任何保证，阿里云不对由此引发的任何损害承担责任。
您应自觉遵守第三方模型“Stable Diffusion”的用户协议、使用规范和相关法律法规，并就使用第三方模型的合法性、合规性自行承担相关责任。

随着生成型AI技术能力的提升，大家越来越关注通过AI模型提升研发效率。作为AIGC （AI Gererative Content）领域的知名项目Stable Diffusion , 可以帮助用户快速、准确地生成想要的场景及图片。但目前使用Stable Diffusion会面临如下问题：

单个Pod处理请求的吞吐率有限，如果多个请求转发到同一个Pod，会导致服务端过载异常，因此需要精准的控制单个Pod请求并发处理数。
由于GPU资源很珍贵，期望做到按需使用资源，在业务低谷及时释放GPU资源。

基于以上两个问题，阿里云容器服务提供Knative解决方案，可以做到基于并发请求数精准处理，实现自动弹性扩缩容，打造生产可用的Stable Diffusion服务。具体实现流程如下所示。

步骤一：部署Stable Diffusion服务

登录容器服务管理控制台，在左侧导航栏选择集群。
在集群列表页面，单击目标集群名称，然后在左侧导航栏，选择应用 > Knative。

在Knative页面的服务管理页签下，选择命名空间为default，然后单击使用模板创建，将以下YAML示例粘贴至模板，最后单击创建，创建一个名为knative-sd-demo的服务。

apiVersion: serving.knative.dev/v1
kind: Service
metadata:
  name: knative-sd-demo
  annotations:
    serving.knative.dev.alibabacloud/affinity: "cookie"
    serving.knative.dev.alibabacloud/cookie-name: "sd"
    serving.knative.dev.alibabacloud/cookie-timeout: "1800"
spec:
  template:
    metadata:
      annotations:
        autoscaling.knative.dev/class: mpa.autoscaling.knative.dev
        autoscaling.knative.dev/maxScale: '10'
        autoscaling.knative.dev/targetUtilizationPercentage: "100"
        k8s.aliyun.com/eci-use-specs: ecs.gn5-c4g1.xlarge,ecs.gn5i-c8g1.2xlarge,ecs.gn5-c8g1.2xlarge  
    spec:
      containerConcurrency: 1
      containers:
      - args:
        - --listen
        - --skip-torch-cuda-test
        - --api
        command:
        - python3
        - launch.py
        image: yunqi-registry.cn-shanghai.cr.aliyuncs.com/lab/stable-diffusion@sha256:62b3228f4b02d9e89e221abe6f1731498a894b042925ab8d4326a571b3e992bc
        imagePullPolicy: IfNotPresent
        ports:
        - containerPort: 7860
          name: http1
          protocol: TCP
        name: stable-diffusion
        readinessProbe:
          tcpSocket:
            port: 7860
          initialDelaySeconds: 5
          periodSeconds: 1
          failureThreshold: 3

如图所示，表明knative-sd-demo服务已部署成功。

步骤二：访问服务

在服务管理页签，获取服务的访问网关和默认域名。
将knative-sd-demo服务的网关地址与需要访问的域名进行Host绑定，在Hosts文件中添加绑定信息。绑定示例如下：
```
47.xx.xxx.xx knative-sd-demo.default.example.com # 网关IP和域名请以您的实际数据为准。
```
完成Host绑定后，在服务管理页签，单击knative-sd-demo服务的默认域名，访问Stable Diffusion。
如图所示，可通过域名直接对Stable Diffusion进行访问。

步骤三：基于请求实现自动弹性扩缩容

使用Hey压测工具，执行压测。

说明

Hey压测工具的详细介绍，请参见Hey。

# 发送50个请求，并发数为5，请求超时时间为180秒。
./hey -n 50 -c 5 -t 180 -m POST -T "application/json"  -d '{"prompt": "pretty dog"}' http://knative-sd-demo.default.example.com/sdapi/v1/txt2img

预期输出：

Summary:
  Total:	252.1749 secs
  Slowest:	62.4155 secs
  Fastest:	9.9399 secs
  Average:	23.9748 secs
  Requests/sec:	0.1983


Response time histogram:
  9.940 [1]	|■■
  15.187 [17]	|■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■
  20.435 [9]	|■■■■■■■■■■■■■■■■■■■■■
  25.683 [11]	|■■■■■■■■■■■■■■■■■■■■■■■■■■
  30.930 [1]	|■■
  36.178 [1]	|■■
  41.425 [3]	|■■■■■■■
  46.673 [1]	|■■
  51.920 [2]	|■■■■■
  57.168 [1]	|■■
  62.415 [3]	|■■■■■■■


Latency distribution:
  10% in 10.4695 secs
  25% in 14.8245 secs
  50% in 20.0772 secs
  75% in 30.5207 secs
  90% in 50.7006 secs
  95% in 61.5010 secs
  0% in 0.0000 secs

Details (average, fastest, slowest):
  DNS+dialup:	0.0424 secs, 9.9399 secs, 62.4155 secs
  DNS-lookup:	0.0385 secs, 0.0000 secs, 0.3855 secs
  req write:	0.0000 secs, 0.0000 secs, 0.0004 secs
  resp wait:	23.8850 secs, 9.9089 secs, 62.3562 secs
  resp read:	0.0471 secs, 0.0166 secs, 0.1834 secs

Status code distribution:
  [200]	50 responses

可以看到持续发送了50个请求，请求成功率为100%。

执行如下命令，可实时观察Pod扩缩容情况。
```
watch -n 1 'kubectl get po'
```
由于部署Stable Diffusion服务时配置了单Pod的最大并发数是1 （containerConcurrency: 1），因此压测期间自动扩容了5个Pod。

步骤四：查看Stable Diffusion服务监控数据

Knative提供开箱即用的可观测能力，在Knative页面，单击监控大盘页签，即可查看Stable Diffusion服务的监控数据情况。如何开启Knative监控大盘，请参见通过阿里云Prometheus监控查看Knative大盘。

在Overview (average over the selected time range)区域，可查看Stable Diffusion服务的请求量（Request Volume）、请求成功率（Success Rate）、4xx（客户端错误）、5xx（服务器端错误）和Pod扩缩容趋势的监控数据。
在Response Time区域，查看Knative的响应延迟数据，包括P50、P90、P95和P99。