跳到正文
原文
Hugging Face Blog·· 2023-01-26精选AI 评分75

用 LoRA 高效微调 Stable Diffusion

Using LoRA for Efficient Stable Diffusion Fine-Tuning

AI 导读

Hugging Face 在 diffusers 中正式支持 LoRA 微调 Stable Diffusion,训练脚本最低可在 11 GB 显存上运行,无需 8-bit 优化器等技巧。由于原模型权重被冻结,只需保存新注入层的权重,单个文件约 3 MB,比原 UNet 小约一千倍,便于分享和下载。文章给出了完整的训练命令、推理时加载 LoRA 权重的代码,以及结合 Dreambooth 的用法。

推荐理由

原文给出 LoRA 微调 Stable Diffusion 的完整脚本与推理流程,读者可据此在 11 GB 显存上复现训练。

来源:Hugging Face Blog · huggingface.co