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

5 R5 C! |+ R# Y  v4 y
4 a/ X2 s3 o2 c; XStable Diffusion安装GitHub
在安装 Stable Diffusion 模型时，通常有两种主要方式：直接从官方 GitHub 仓库克隆代码，或通过 Hugging Face 等平台安装预训练模型。下面将分别介绍这两种方法。

  ^1 u/ o+ I/ ?1 s# w7 H6 Z. }方法一：从 GitHub 克隆代码
) P1 s# Q8 Y$ k0 g! X) P克隆 Stable Diffusion 仓库
7 \0 `8 p3 C0 s+ q  W8 t
$ F0 O! S- m8 z8 p6 V; g首先，你需要从 GitHub 克隆 Stable Diffusion 的官方仓库。截至目前（2023年），官方仓库地址为：

: n, e; W3 ~8 q! G9 g' R% K3 A7 S/ Ihttps://github.com/CompVis/stable-diffusion

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

git clone https://github.com/CompVis/stable-diffusion.git
cd stable-diffusion

设置环境
$ c! h- b  S1 c  L. f
安装必要的依赖。Stable Diffusion 需要 Python 和一些特定的库。你可以使用 requirements.txt 文件中的依赖来安装它们。首先，确保你的 Python 版本符合要求（通常是 Python 3.8 或更高）。然后，使用以下命令安装依赖：

pip install -r requirements.txt
$ [  y- ]8 W  `$ L
3 ~4 p; r3 j- m( |7 u7 _运行 Stable Diffusion
' Y: K* {2 V& O  i$ q# ^
根据你的具体需求，你可以运行不同的脚本。例如，如果你只是想测试模型，可以运行：
+ @& P9 h; K4 x3 L* F* B
# K3 M8 t$ v& C% Ppython scripts/txt2img.py --prompt "a photo of an astronaut riding a horse"

方法二：通过 Hugging Face 安装预训练模型
, @9 j& x. z; E2 O& Q' `9 h  s如果你只是想使用预训练的模型而不想从源代码开始，你可以使用 Hugging Face 的 diffusers 库来加载和使用 Stable Diffusion 模型。
( ~" z# m) E: [1 h/ c! {
+ {9 V6 z7 w: R) K9 q! v1 _, M安装 diffusers
4 t0 U% \8 E& K5 @; {  y
9 L, }7 p8 U$ m3 u  I7 ~首先，安装 diffusers 库：

pip install diffusers transformers scipy

加载和使用模型
- u* H. Z& ^; k( x
使用以下代码加载和使用 Stable Diffusion 模型：

from diffusers import StableDiffusionPipeline
* s0 U! E3 a: Z4 c. ?7 I& Ximport torch
. j9 u- m" y% h9 ^( R6 o% e
device = "cuda" if torch.cuda.is_available() else "cpu"
% P% Z7 G" w" o# Smodel_id = "runwayml/stable-diffusion-v1-5"  # 使用官方预训练模型ID
pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
pipe = pipe.to(device)
: s& Z& D4 G, ?( Z
  ~  q# S, |" D  kimage = pipe("a photo of an astronaut riding a horse", guidance_scale=7.5)
image[0].save("astronaut_riding_horse.png")
9 h, C$ r) |1 k  W: s  B; K
* j0 @# M$ z& z9 }! r/ ~  A: z" ]' U确保替换 use_auth_token=True 中的 True 为你的 Hugging Face 访问令牌，如果你有的话。如果没有，可以省略或设置为 None。你可以在 Hugging Face 的网站上获取访问令牌。
* ^! U8 t; J' c% {* Q% F3 p$ _, Y
; y0 f) Y; u5 O2 S- F1 T4 @' `/ t通过上述任何一种方法，你都可以开始使用 Stable Diffusion 生成图像了。选择哪种方法取决于你的具体需求，比如是否需要自定义模型或只是快速开始使用预训练模型。

  I4 C) a! x' ?3 p; z7 r提示：AI自动生成，仅供参考
# g5 B7 `0 T$ |. ?/ N
参考