AIGC - Stable Diffusion 搭建【从零到一】

AIGC - Stable Diffusion 搭建

提示：这里可以添加系列文章的所有文章的目录，目录需要自己手动添加
例如：第一章 Python 机器学习入门之pandas的使用

---

提示：写完文章后，目录可以自动生成，如何生成可参考右边的帮助文档

---

1. Huggingface Token

url = https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/stable_diffusion.ipynb#scrollTo=gd-vX3cavOCt
1. 注册获取账号
2. 生成获取用户项目token

2. 租用GPU

url = https://matpool.com/ # 矩池云
1. 注册获取账号
2. 选择GPU档次及预装型号（建议huggingface,Pytorch）
3. 建立Pycharm关联（SSH）
4. JupyterLab 
   pip3 install diffusers 
   pip3 install accelerate
   pip3 install spacy ftfy==4.4.3

3. Pycharm

import diffusers,torch

prompt   = "illustration of a piano with a boy playing"
model_id = "CompVis/stable-diffusion-v1-4"
device   = torch.device("cuda")
pipe     = diffusers.StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
pipe     = pipe.to(device)
image    = pipe(prompt).images[0]
image.save("ai_image.png")