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

Hugging Face 发布 diffusers 中 LoRA 用于 Stable Diffusion 高效微调的指南

Using LoRA for Efficient Stable Diffusion Fine-Tuning

AI 导读

Hugging Face 在 diffusers 库中正式支持使用 LoRA 技术对 Stable Diffusion 进行高效微调。该方法通过冻结预训练权重并注入可训练层,使全量微调仅需 11 GB VRAM 即可运行,且生成的权重文件仅约 3 MB,比原始 UNet 模型小约一千倍。

推荐理由

原文提供了在 diffusers 中使用 LoRA 进行 Stable Diffusion 微调的具体脚本和推理代码,展示了显著降低显存需求和模型文件大小的实际效果。

来源:Hugging Face Blog · huggingface.co