Stable-diffusion-WebUI 的API调用(内含文生图和图生图实例)

前情提要

在之前尝试使用Diffusers库来进行stable-diffusion的接口调用以及各种插件功能实现，但发现diffusers库中各复杂功能的添加较为麻烦，而且难以实现对采样器的添加，safetensors格式模型的读取。在官网上找到了webui有专门的api接口，能够极大方便我们进行类似webui界面的api调用。

diffusers文档

webui项目官网

webui API说明

webui项目部署

这种调用webui自带的api的方法需要先将webui运行起来，无论是自己从官网配置的webui，还是各类启动器一键启动的都是可以的。（我使用的为一键启动包，较为简单）

一键启动包教程

如果是自己配置的

使用

bash webui.sh --nowebui

或者

python launch.py --xformers --api

API接口调用

当我们把webui项目启动之后，我们可以看到运行的端口（默认为7860）

可以进行调用

1. 文生图（python示例）：

1. import json
2. import requests
3. import io
4. import base64
5. from PIL import Image
7. url = "http://127.0.0.1:7860"
9. prompt = "dog"
10. negative_prompt = ""
12. payload = {
14. # 模型设置
15. "override_settings":{
16. "sd_model_checkpoint": "v1-5-pruned.ckpt",
17. "sd_vae": "animevae.pt",
18. "CLIP_stop_at_last_layers": 2,
19. },
21. # 基本参数
22. "prompt": prompt,
23. "negative_prompt": negative_prompt,
24. "steps": 30,
25. "sampler_name": "Euler a",
26. "width": 512,
27. "height": 512,
28. "batch_size": 1,
29. "n_iter": 1,
30. "seed": 1,
31. "CLIP_stop_at_last_layers": 2,
33. # 面部修复 face fix
34. "restore_faces": False,
36. #高清修复 highres fix
37. # "enable_hr": True,
38. # "denoising_strength": 0.4,
39. # "hr_scale": 2,
40. # "hr_upscaler": "Latent",
42. }
44. response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
45. r = response.json()
46. image = Image.open(io.BytesIO(base64.b64decode(r['images'][0])))
48. image.show()
49. image.save('output.png')

2. 图生图（python 示例）

1. import json
2. import requests
3. import io
4. import base64
5. from PIL import Image
6. import cv2
8. url = "http://127.0.0.1:7860"
10. prompt = "cat"
11. negative_prompt = ""
13. # 此处为读取一张图片作为输入图像
14. img = cv2.imread('image.jpg')
16. # 编码图像
17. retval, bytes = cv2.imencode('.png', img)
18. encoded_image = base64.b64encode(bytes).decode('utf-8')
20. payload = {
22. # # 模型设置
23. # "override_settings":{
24. # "sd_model_checkpoint": "v1-5-pruned.ckpt",
25. # "sd_vae": "animevae.pt",
26. # "CLIP_stop_at_last_layers": 2,
27. # },
29. # 基本参数
30. "prompt": prompt,
31. "negative_prompt": negative_prompt,
32. "steps": 30,
33. "sampler_name": "Euler a",
34. "width": 512,
35. "height": 512,
36. "batch_size": 1,
37. "n_iter": 1,
38. "seed": 1,
39. "cfg_scale": 7,
40. "CLIP_stop_at_last_layers": 2,
42. "init_images": [encoded_image],
44. # 面部修复 face fix
45. "restore_faces": False,
47. #高清修复 highres fix
48. # "enable_hr": True,
49. # "denoising_strength": 0.4,
50. # "hr_scale": 2,
51. # "hr_upscaler": "Latent",
53. }
57. response = requests.post(url=f'{url}/sdapi/v1/img2img', json=payload)
58. r = response.json()
59. image = Image.open(io.BytesIO(base64.b64decode(r['images'][0])))
61. image.show()
62. image.save('output.png')

如要修改其他参数可参照官网文档进行修改。