# AI绘画stable diffusion 实践

## 1.创建一台GPU云主机
创建流程参考[创建第一台云主机](/docs/uhost/newuser/briefguide)   
创建GPU云主机时，镜像选择“AI绘画stable diffusion平台”，操作路径：镜像市场——>AI绘画stable diffusion平台，便捷安装stable diffusion，镜像内置环境：CentOS 7.8。   
推荐机型及配置：   
GPU型云主机 T4/T4S 最低配置：8核32G一颗GPU    
GPU型云主机 V100S  最低配置：10核32G一颗GPU   
GPU型云主机 P40    最低配置：8核32G 一颗GPU   

> 内存请选择32GB及以上，否则模型加载时可能会触发OOM。 
绑定EIP并在外网防火墙放行TCP 8888端口。
   
 ## 2.使用stable diffusion
 ### 2.1 方式一：使用jupyter nodebook新建ldm环境（推荐）
 根据外网ip地址，访问jupyter：http://EIP:8888
 输入token（在/root/.jupyter/jupyter_notebook_config.py中查看c.NotebookApp.token的配置，可自行修改。）
 
 新建ldm
 
如您使用**英文版**描述，可参考以下示例：

添加如下示例代码后运行（以"A thriving view alongside Pearl of the Orient in Shanghai , by Van Gogh, oil painting trending on artstation HQ"为示例）
```
from torch import autocast
from diffusers import StableDiffusionPipeline
from IPython.display import Image
 
width = 512
height = 512
 
pipe = StableDiffusionPipeline.from_pretrained(
    "/root/demo/stable-diffusion-v1-4").to("cuda")
 
prompt = "A thriving view alongside Pearl of the Orient in Shanghai , by Van Gogh, oil painting trending on artstation HQ"
with autocast("cuda"):
    image = pipe(prompt, height, width)["sample"][0] 
     
image.save("Van_Gogh_Style_Shanghai.png")
 
# show the image in web
Image(filename = 'Van_Gogh_Style_Shanghai.png', width=width, height=height)
```
生成图片即可立即查看。


如您使用**中文版**描述，可参考以下示例：

添加如下示例代码后运行（以"大漠孤烟直，长河落日圆，油画"为示例）
```
from diffusers import StableDiffusionPipeline
from IPython.display import Image

width = 512
height = 512

pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1").to("cuda")

prompt = '大漠孤烟直，长河落日圆，油画'
image = pipe(prompt, guidance_scale=7.5).images[0]  
image.save("油画.png")

# show the image in web
Image(filename = '油画.png', width=width, height=height) 
```
生成图片即可立即查看。


> 1. 选中代码分区，点击【运行】，如出现In[* ]，则表示代码运行中，静等出图结果即可；   
> 2. 如您需要调整画布尺寸，需保证调整后长宽的均为8的整数倍。

**如您想快速尝试，还可直接使用demo/路径下的SD_Demos.ipynb，内置英文版、中文版模型**


 
### 2.2 方式二：使用stable diffusion的sample script
#### 2.2.1 切换conda环境

```
conda activate ldm
```
#### 2.2.2 执行sample脚本
执行脚本，输入您预想图画的描述，即可得到图片（以下以“A valley full of flowers , by Van Gogh, oil painting trending on artstation HQ”为例），生成的图片在 /root/stable-diffusion/outputs/txt2img-samples/目录下。
```
cd stable-diffusion
python scripts/txt2img.py --prompt "A valley full of flowers , by Van Gogh, oil painting trending on artstation HQ"
```
#### 2.2.3 使用jupyter页面查看
根据外网ip地址，访问jupyter：http://EIP:8888

输入token（在/root/.jupyter/jupyter_notebook_config.py中查看c.NotebookApp.token的配置，可自行修改。）

根据导航点击预览图片