【玩机教程】Stable Diffusion 保姆级教程|H

🔥Stable Diffusion 保姆级教程🔥

### 配置 Hugging Face 国内镜像
设置国内镜像以解决下载模型超时、失败问题,加速下载 Stable Diffusion、LLM 等模型。

#### 不同系统配置方法
– **Windows 系统(CMD 命令行)**
“`bash
set HF_ENDPOINT=https://hf-mirror.com
“`
– **Windows 系统(PowerShell)**
“`powershell
$env:HF_ENDPOINT=”https://hf-mirror.com”
“`
– **Mac/Linux 系统**
“`bash
export HF_ENDPOINT=https://hf-mirror.com
“`
#### 永久配置(推荐)
– **Windows**
右键点击“此电脑”→“属性”→“高级系统设置”→“环境变量”,新建用户变量:
– 变量名:HF_ENDPOINT
– 变量值:https://hf-mirror.com
– **Mac/Linux**
“`bash
echo ‘export HF_ENDPOINT=https://hf-mirror.com’ >> ~/.bashrc
source ~/.bashrc
“`

### 一键安装命令
– **CPU 版本**
“`bash
pip install diffusers transformers accelerate torch safetensors pillow
“`
– **GPU 版本(需 NVIDIA GPU)**
“`bash
pip install –pre torch torchvision torchaudio –index-url https://download.pytorch.org/whl/nightly/cu132
pip install diffusers transformers accelerate safetensors pillow
“`

### 补充依赖
根据需求安装:
“`bash
pip install modelscope huggingface_hub diffusers torch transformers
“`

### 模块作用
– **diffusers**: Hugging Face 官方扩散模型库,用于图像生成。
– **transformers**: 运行大语言模型和 CLIP 文本编码器的核心库。
– **accelerate**: 自动加速 PyTorch 模型,支持 GPU/TPU、多卡训练。
– **torch**: PyTorch 深度学习框架,AI 模型的底层运行环境。
– **safetensors**: 安全、快速的模型权重存储格式。
– **pillow**: Python 图像处理库,用于打开、保存、显示图片。

### 最小可运行代码(文生图)
“`python
from diffusers import StableDiffusionPipeline
import torch

model_id = “runwayml/stable-diffusion-v1-5”
pipe = StableDiffusionPipeline.from_pretrained(
model_id,
torch_dtype=torch.float16,
use_safetensors=True
).to(“cuda”)

prompt = “a cute cat running on the grass, 8k, realistic”
image = pipe(prompt).images[0]
image.save(“cat.png”)
print(“图片已保存:cat.png”)
“`

### 关键概念补充
– **StableDiffusionPipeline**: 端到端的图像生成管道。
– **torch_dtype=torch.float16**: 使用 FP16 精度,减少显存占用。
– **use_safetensors**: 使用 safetensors 格式,比 pickle 更安全。
– **.to(“cuda”)**: 在有 NVIDIA GPU 的环境中运行。

### 文件拆分
– **download_model.py**: 下载并保存模型。
– **generate_image.py**: 从本地加载并生成图片。
– **sd_generate.py**: 智能判断(自动下载或本地加载)。

© 版权声明
THE END
喜欢就支持以下吧
点赞25赞赏 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容