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