使用stable diffussion插件StableSR将图片高清放大

一：需要安装的插件

1、StableSR，项目地址：https://github.com/pkuliyi2015/sd-webui-stablesr

不过国内没什么用，访问不了，可以用下面的国内镜像：

https://gitee.com/han51535/sd-webui-stablesr.git

这个试过，还是可以用的。

2、安装“Multidiffusion”插件

项目地址：https://github.com/pkuliyi2015/multidiffusion-upscaler-for-automatic1111

当然同样访问不了，我们还是用国内的镜像替代

https://gitee.com/stable_diffusion/multidiffusion-upscaler-for-automatic1111.git

二：需要下载的模型

1、StabilityAI”的官方大模型（5.21GB）

名称：v2-1_768-ema-pruned.ckpt

下载地址：https://huggingface.co/stabilityai/stable-diffusion-2-1/tree/main

安装位置：stable-diffusion-webui/models/Stable-Diffusion/ 文件夹中

2、“VAE”模型。（745MB）

名称：vqgan_cfw_00011_vae_only.ckpt（约750MB大小）

下载地址：https://drive.google.com/file/d/1ARtDMia3_CbwNsGxxGcZ5UP75W4PeIEI/view

放在你的 stable-diffusion-webui/models/VAE

3、“StableSR”脚本模型。（400MB）

名称：webui_768v_139.ckpt

下载地址：https://huggingface.co/Iceclear/StableSR/blob/main/webui_768v_139.ckpt

放置位置：/stable-diffusion-webui/extensions/sd-webui-stablesr/models

虽然都访问不到，但是还是列出名称和链接，万一有会魔法的小伙伴能用到呢。

这里我上传了这些模型，找了整整一个晚上，算是东拼西凑的弄完整了。实在找不到的小伙伴可以从这里下载。

百度网盘下载：https://pan.baidu.com/s/1HH9RQihfFOa6kiIBguvpyg?pwd=lgpk 提取码：lgpk 

三：开始工作

1、打开SD WEBUI，转到图生图，打开测试的图片（下载包里有一张），也可以自己找。

2、Tiled Diffusion设置

首先要启动VAE切片，前要要打勾

编码和解码切片大小不要设置太大，会占用较大的内存。我这台机器的是P40，24G的GPU，目前设置的值仅供参考。

3、脚 本设置

拉到页面最下面，脚本处。 

选择StableSR脚本模型

SR Model：选择webui_768_139.ckpt，缩放比，这里选的是2，就是放大两倍。

其他可选默认值

点击生成

四：问题解答：

生成过程中内存溢出的解决方法，打开stable diffusion项目目录

vim webui.py

if__name__=="__main__":
#在这句的下面加上下面这句话
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

然后再执行启动stable diffusion

python webui.py

如果你是使用launch.py启动的，也一样，把这句加到那里。只是你需要在launch.py最前面加上

import torch

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')