1.2 在MacBook本地运行Stable Diffusion-ChatGPT与AIGC工具入门实战指南

当前位置:　首页>> 技术小册>> ChatGPT与AIGC工具入门实战指南

1.2 在MacBook本地运行Stable Diffusion

引言

Stable Diffusion作为近年来最热门的文本到图像生成模型之一，其基于Transformer结构的强大能力使得它能够在给定的文本描述下生成高度逼真且富有创意的图像。对于创意工作者、艺术家、设计师以及AI爱好者而言，能够在本地环境中部署并运行Stable Diffusion，不仅意味着更高的灵活性和自主性，还能极大地提升工作效率。本章节将详细指导你如何在MacBook上安装并运行Stable Diffusion，让你轻松开启AI绘画之旅。

准备工作

在开始之前，请确保你的MacBook满足以下基本要求：

操作系统：macOS Catalina（10.15）及以上版本。
硬件要求：由于Stable Diffusion计算量较大，建议MacBook配备至少8GB RAM，更推荐使用配备独立显卡（如M1 Pro/Max芯片或Intel Iris Plus/AMD Radeon系列）的型号。
存储空间：预留至少20GB的硬盘空间用于安装相关软件和模型。
Python环境：安装最新版本的Python（推荐3.8及以上），并配置好pip。

步骤一：安装必要的软件与库

安装Anaconda（可选但推荐）
Anaconda是一个流行的Python数据科学和机器学习平台，它简化了包管理和部署。你可以从Anaconda官网下载并安装适用于macOS的版本。安装完成后，打开终端（Terminal），通过conda命令创建一个新的Python环境，并安装必要的库。
```
conda create -n stable-diffusion python=3.8
conda activate stable-diffusion
```
安装PyTorch
Stable Diffusion依赖于PyTorch进行深度学习计算。根据你的MacBook配置（CPU或GPU），选择适合的PyTorch版本进行安装。以下命令以安装支持CUDA的PyTorch为例（如果你的Mac没有NVIDIA GPU，则可以选择安装CPU版本的PyTorch）。
```
# 对于M1芯片Mac，你可能需要使用miniforge或conda-forge的PyTorch版本
# 这里以CPU版本为例
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
安装其他依赖库
Stable Diffusion还需要其他如transformers、torchvision等库。你可以通过pip安装它们：
```
pip install transformers torchvision
```
此外，你可能还需要安装omegaconf、PIL（Pillow）、matplotlib等库用于配置管理、图像处理和可视化。
```
pip install omegaconf pillow matplotlib
```

步骤二：获取Stable Diffusion模型与权重

下载Stable Diffusion仓库
访问Stable Diffusion GitHub仓库或其衍生项目的GitHub页面（如stable-diffusion-webui），克隆仓库到本地。
```
git clone https://github.com/CompVis/stable-diffusion.git
cd stable-diffusion
```
下载预训练模型
根据仓库的README或文档，下载Stable Diffusion的预训练模型及其权重文件。这些文件通常较大，可能需要较长时间下载。确保下载到正确的版本，并放置在项目的指定文件夹中。

步骤三：配置与运行Stable Diffusion

配置环境
根据下载的Stable Diffusion版本，可能需要编辑配置文件（如YAML格式的配置文件）以设置模型路径、输出目录等。确保所有路径都正确无误。
运行Stable Diffusion
在终端中，使用Python运行Stable Diffusion的启动脚本。这通常是一个Python脚本文件，如run_diffusion.py（具体名称可能因版本而异）。运行前，请确保你的Python环境已激活，并位于Stable Diffusion项目的根目录下。
```
python scripts/run_diffusion.py --prompt "a beautiful sunset over the mountains" --output_dir ./outputs
```
上述命令将使用“a beautiful sunset over the mountains”作为文本提示，生成图像并保存到./outputs目录。
使用Web界面（可选）
如果你更喜欢图形用户界面（GUI），可以考虑使用如stable-diffusion-webui这样的项目，它提供了一个基于Web的交互界面，使得操作更加直观方便。安装和配置方法类似于上述步骤，但通常包含额外的Web服务器设置。