Modal Blog· Yiren Lu·· 2024-05-21AI 评分31
在 Heroicons 上微调 Stable Diffusion,打造无限图标库
Create an infinite icon library by fine-tuning Stable Diffusion
AI 导读
Modal 在约 300 个 SVG 的 Heroicons 图标库上微调了一个 Stable Diffusion 模型,以生成该风格的新黑白线条图标,包括 barack-obama 和 golden-retriever 等概念的图标。完整流程——下载预训练模型、使用 HuggingFace Diffusers 的 train_text_to_image.py 脚本进行全量微调,以及提供 Gradio Web UI——都在 Modal 的无服务器 GPU 平台上运行。文章详细介绍了数据集准备(SVG 转 PNG、白色背景、caption metadata.csv),并指出全量微调效果最好,不过 LoRA 风格的并行适配器也可用于小数据集。
来源:Modal Blog · modal.com