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

Stable Diffusion安装GitHub
  `6 Q! c2 o3 l! O在安装 Stable Diffusion 模型时，通常有两种主要方式：直接从官方 GitHub 仓库克隆代码，或通过 Hugging Face 等平台安装预训练模型。下面将分别介绍这两种方法。

* V. U  {. H4 M' D! s$ N* B0 b方法一：从 GitHub 克隆代码
8 n6 ?" z$ l/ L; W) ]1 V4 p5 J/ I克隆 Stable Diffusion 仓库
% \1 X( J9 D: B9 M2 q% I
首先，你需要从 GitHub 克隆 Stable Diffusion 的官方仓库。截至目前（2023年），官方仓库地址为：

https://github.com/CompVis/stable-diffusion

在终端中，使用以下命令克隆仓库：

3 `" @8 B- e( ]5 D- C# l  pgit clone https://github.com/CompVis/stable-diffusion.git
cd stable-diffusion

设置环境
; J/ M$ S& L/ I4 j% U1 y1 O
3 F' P# {1 p1 d3 ~2 M/ x安装必要的依赖。Stable Diffusion 需要 Python 和一些特定的库。你可以使用 requirements.txt 文件中的依赖来安装它们。首先，确保你的 Python 版本符合要求（通常是 Python 3.8 或更高）。然后，使用以下命令安装依赖：

6 f* m, j$ _7 Jpip install -r requirements.txt
1 C1 v- T" `, E7 v( d/ {
1 L8 Y7 z' J) w运行 Stable Diffusion

# ^; a$ C- I8 `! U' r- i* W% }根据你的具体需求，你可以运行不同的脚本。例如，如果你只是想测试模型，可以运行：
/ h' y0 W& `7 ~8 v
python scripts/txt2img.py --prompt "a photo of an astronaut riding a horse"

! d7 ~5 w, R7 x, ]5 ^4 j2 l方法二：通过 Hugging Face 安装预训练模型
如果你只是想使用预训练的模型而不想从源代码开始，你可以使用 Hugging Face 的 diffusers 库来加载和使用 Stable Diffusion 模型。

安装 diffusers

首先，安装 diffusers 库：
& X8 E1 h) \  D& a! v8 v
pip install diffusers transformers scipy

- q" |$ x9 R0 f9 L加载和使用模型

使用以下代码加载和使用 Stable Diffusion 模型：
/ B. g4 ~& i( Q+ c7 H; z; q- v
# m! Y+ ]. I8 `! K! Z: Efrom diffusers import StableDiffusionPipeline
1 z/ Z% P( `5 e+ s1 A4 p3 nimport torch

) ~- ^7 P! o) W; adevice = "cuda" if torch.cuda.is_available() else "cpu"
! C9 B& k- j+ M: V5 M3 \model_id = "runwayml/stable-diffusion-v1-5"  # 使用官方预训练模型ID
! e, w! E: b# S& Q5 ]" J1 ypipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
+ ~: m2 i& ]# Z; ]' zpipe = pipe.to(device)
; v, Q* ]  }7 W4 O5 }, x$ n! E8 q
6 K4 A& y6 H0 u4 I3 @/ y# F% limage = pipe("a photo of an astronaut riding a horse", guidance_scale=7.5)
6 w: `$ w* M2 Q1 [2 aimage[0].save("astronaut_riding_horse.png")
1 |" a6 P% n9 o6 K! F6 t
确保替换 use_auth_token=True 中的 True 为你的 Hugging Face 访问令牌，如果你有的话。如果没有，可以省略或设置为 None。你可以在 Hugging Face 的网站上获取访问令牌。
) g9 [, p0 F6 V& x4 J7 t; L$ J
通过上述任何一种方法，你都可以开始使用 Stable Diffusion 生成图像了。选择哪种方法取决于你的具体需求，比如是否需要自定义模型或只是快速开始使用预训练模型。

5 K. u, ]8 G" a; ~& ^' \7 y. Q提示：AI自动生成，仅供参考

参考