pytorch-stable-diffusion

⚠ 该项目 README 未提供完整安装说明，以下步骤为通用流程，请以官方文档为准。

环境要求

• 能运行 Jupyter Notebook 的本地环境（README 未给出具体依赖安装步骤）
• Git（用于克隆仓库，README 未显式说明）
• 足够的磁盘空间存放 stable-diffusion-v1-5 的权重文件
• 可访问 Hugging Face 网站以下载模型与分词器文件

安装与启动步骤

1. 1克隆项目仓库

   从 GitHub 获取源码，README 未提供安装命令，此步为通用准备动作，需自行准备 Jupyter 环境。

   git clone https://github.com/hkproj/pytorch-stable-diffusion.git

   复制

2. 2进入项目目录

   切换到仓库根目录，后续下载的权重与分词器文件都要放在其 data 目录下。

   cd pytorch-stable-diffusion

   复制

3. 3创建 data 目录

   README 要求把文件保存到 data 文件夹，若目录不存在需先创建，注意大小写与路径。

   mkdir -p data

   复制

4. 4下载分词器文件

   打开 stable-diffusion-v1-5 的 tokenizer 页面，下载 vocab.json 和 merges.txt 两个文件，保存到 data 文件夹。

5. 5下载模型权重

   从 stable-diffusion-v1-5 仓库主页面下载 v1-5-pruned-emaonly.ckpt，保存到 data 文件夹，文件较大请耐心等待。

6. 6（可选）下载微调权重

   README 测试过任意不超过 v1.5 的微调 ckpt，如 InkPunk Diffusion、Illustration Diffusion，下载后同样放入 data。

如何确认成功

README 未给出启动命令，确认 data 目录下已存在 vocab.json、merges.txt、v1-5-pruned-emaonly.ckpt 三个文件即完成准备。

常见问题

注意事项

• README 未提供任何 pip/conda 依赖安装命令，运行前请自行确认 Python 与 PyTorch 环境可用。
• 权重文件为 Hugging Face 网页链接，需手动下载或用浏览器/命令行工具获取，README 未给出直链命令。
• 除官方 v1-5 权重外，README 还验证过 InkPunk Diffusion 与 Illustration Diffusion 两类微调模型。