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 網頁介面——都在 Modal 的無伺服器 GPU 平台上執行。文章詳細說明了資料集準備(SVG 轉 PNG、白色背景、標註檔 metadata.csv),並指出完整微調效果最好,不過 LoRA 式的平行適配器也可用於小型資料集。
來源:Modal Blog · modal.com