models/sd2.1_base_paddle/conversion_guide.md

SD2.1基础模型转换说明

转换状态

• 源文件: v2-1_512-nonema-pruned.safetensors (PyTorch格式)
• 目标文件: model.pdparams (PaddlePaddle格式)
• 文件大小: 4.9GB
• 状态: 待转换

转换方法

方法1: 使用diffusers库

from diffusers import StableDiffusionPipeline
import torch

# 加载PyTorch模型
pipe = StableDiffusionPipeline.from_pretrained(
    "models/sd2.1_base",
    torch_dtype=torch.float32,
    use_safetensors=True
)

# 转换为PaddlePaddle格式
pipe.save_pretrained("models/sd2.1_base_paddle")

方法2: 使用PaddlePaddle官方工具

# 安装转换工具
pip install paddlepaddle
pip install paddlenlp

# 使用转换脚本
python -m paddlenlp.transformers.convert_pytorch_checkpoint_to_paddle \
    --model_name_or_path models/sd2.1_base \
    --output_dir models/sd2.1_base_paddle

注意事项

• 确保有足够的磁盘空间
• 转换过程可能需要较长时间
• 建议在转换前备份原始文件
• 验证转换后的模型功能