IT评测·应用市场-qidao123.com

标题: Stable-diffusion-WebUI 的API调用(内含文生图和图生图实例) [打印本页]
作者: 前进之路    时间: 2024-7-17 15:48
标题: Stable-diffusion-WebUI 的API调用(内含文生图和图生图实例)
前情提要

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


diffusers文档
webui项目官网
webui API阐明
webui项目部署

这种调用webui自带的api的方法必要先将webui运行起来，无论是自己从官网配置的webui，还是各类启动器一键启动的都是可以的。（我使用的为一键启动包，较为简单）
一键启动包教程
如果是自己配置的
使用
  1. bash webui.sh --nowebui
复制代码
大概
  1. 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,
  41.    
  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')
复制代码
如要修改其他参数可参照官网文档举行修改。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。



欢迎光临 IT评测·应用市场-qidao123.com (https://dis.qidao123.com/) Powered by Discuz! X3.4