安装 NovelAI 原版后端（stable diffusion）-Toy模板网

这篇具有很好参考价值的文章主要介绍了安装 NovelAI 原版后端（stable diffusion）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

安装 NovelAI 原版后端#

详见此教程: NovelAI 原版网页UI+后端部署教程

硬件需求#

一台拥有一张至少有 11G 显存的 NVIDIA GPU 的 Linux 系统的 x86 设备。

建议显存低于 24G 不要尝试使用官方版本前后端。

软件需求#

NVIDIA CUDA 驱动 (CUDA Toolkit 11.6)
Docker 19+
nvidia-container-toolkit

安装 Docker#

bash

curl -fsSL https://get.docker.com | bash

# 国内镜像

curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun

驱动相关#

安装显卡驱动

使用 nvidia-smi 查看显卡状态。

使用 docker run --help | grep -i gpus 确认nvidia-container-toolkit安装成功。

运行相关#

容器状态#

查询出容器 ID docker ps

查看容器日志 docker logs [Container ID]

出现 “Application startup complete.” 即代表程序已经就绪

请求 API

参考代码，具体参照leak的前端后端项目，以及其中的 sd-private\hydra-node-http\main.py

prompt 中 masterpiece, best quality, 开头对应原版 Add Quality Tags 选项，不建议删除，后面直接跟自己 prompt 即可

uc 部分对应 Undesired Content，建议保留默认

sampler 是采样方法，可选 plms/ddim/k_euler/k_euler_ancestral/k_heun/k_dpm_2/k_dpm_2_ancestral/k_lms

seed 是种子，自己随机一个整数数字，不然一直会出一样的结果

n_samples 代表要生成几张图片

python文章来源地址https://www.toymoban.com/news/detail-488247.html

import requests
import json
import base64
import random

endpoint = "http://10.10.12.67/generate"
data = {
  "prompt": "masterpiece, best quality, brown red hair,blue eyes,twin tails,holding cat",
  "seed": random.randint(0,2**32),
  "n_samples": 1,
  "sampler": "ddim",
  "width": 512,
  "height": 768,
  "scale": 11,
  "steps": 28,
  "uc": "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
}


req = requests.post(endpoint, json=data).json()
output = req["output"]

for x in output:
  img = base64.b64decode(x)

  with open("output-" + str(output.index(x)) + ".png", "wb") as f:
    f.write(img)

到了这里，关于安装 NovelAI 原版后端（stable diffusion）的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！