适合画图修图的AI模型开源程序

& A" M9 i" r1 p1 }$ F/ o: @9 XStable Diffusion安装GitHub
在安装 Stable Diffusion 模型时，通常有两种主要方式：直接从官方 GitHub 仓库克隆代码，或通过 Hugging Face 等平台安装预训练模型。下面将分别介绍这两种方法。

- t: Z$ v7 t3 R) c. K方法一：从 GitHub 克隆代码
7 h4 f! o1 Q+ R2 F& d克隆 Stable Diffusion 仓库
2 n2 A1 D' N$ U( o
  `! R# W" J: g* o( S+ D首先，你需要从 GitHub 克隆 Stable Diffusion 的官方仓库。截至目前（2023年），官方仓库地址为：
# A7 P$ B) c% _
https://github.com/CompVis/stable-diffusion
- `2 B6 l3 j: g3 j* N' l
( k# P, R+ @8 B  n+ Z) Q在终端中，使用以下命令克隆仓库：
# p; G1 i5 @* C
2 v; c7 x/ e4 Z4 Tgit clone https://github.com/CompVis/stable-diffusion.git
! c# S+ C% @% gcd stable-diffusion

6 ^1 Q6 N& W0 J) C: t; n( |0 A: X设置环境
3 y" {% s8 O5 z0 k  ^
安装必要的依赖。Stable Diffusion 需要 Python 和一些特定的库。你可以使用 requirements.txt 文件中的依赖来安装它们。首先，确保你的 Python 版本符合要求（通常是 Python 3.8 或更高）。然后，使用以下命令安装依赖：

pip install -r requirements.txt
' s* g% g* ^7 x! c
3 \; |( }- C3 \; x运行 Stable Diffusion

根据你的具体需求，你可以运行不同的脚本。例如，如果你只是想测试模型，可以运行：
2 [3 i/ a! s: |/ g' `
python scripts/txt2img.py --prompt "a photo of an astronaut riding a horse"

方法二：通过 Hugging Face 安装预训练模型
( h, g2 m' P) Z* w如果你只是想使用预训练的模型而不想从源代码开始，你可以使用 Hugging Face 的 diffusers 库来加载和使用 Stable Diffusion 模型。
' h2 t7 y  y. A. v
安装 diffusers

" F: W5 F1 C- ~6 A! e( l首先，安装 diffusers 库：
$ l; I' O8 D' Z1 `2 b- J( a- A
/ k  `% X/ ^: D5 \pip install diffusers transformers scipy

1 Q7 [1 v# {/ e' ?# K( E9 h  N加载和使用模型

使用以下代码加载和使用 Stable Diffusion 模型：

  c' H4 X2 t0 t% W' g" Kfrom diffusers import StableDiffusionPipeline
import torch

device = "cuda" if torch.cuda.is_available() else "cpu"
1 a/ U. `4 V( a4 n) v* ~% B8 Smodel_id = "runwayml/stable-diffusion-v1-5"  # 使用官方预训练模型ID
( O  r5 t! _; I# hpipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
" s! }4 l  H. o5 Tpipe = pipe.to(device)

image = pipe("a photo of an astronaut riding a horse", guidance_scale=7.5)
/ I8 N! ^7 T2 H5 {1 @5 R0 G, himage[0].save("astronaut_riding_horse.png")
9 Y0 a+ \) W* r4 U% b' `
确保替换 use_auth_token=True 中的 True 为你的 Hugging Face 访问令牌，如果你有的话。如果没有，可以省略或设置为 None。你可以在 Hugging Face 的网站上获取访问令牌。
3 [( f. f+ D$ Y9 b2 s/ m, H
通过上述任何一种方法，你都可以开始使用 Stable Diffusion 生成图像了。选择哪种方法取决于你的具体需求，比如是否需要自定义模型或只是快速开始使用预训练模型。

# R% \; ?- h  Y, M0 W  M: F& _提示：AI自动生成，仅供参考

: I$ `% {9 F4 v0 U参考