ostris/ai-toolkit

它解決了什麼問題

AI Toolkit 提供一個統一且易於使用的擴散模型訓練套件，使用戶能夠在消費級硬體上訓練圖像、影片和音訊模型，而無需手動管理複雜的環境設置。

它是如何運作的

該工具包支援廣泛的模型（包括 FLUX、SDXL 和 Wan 2.1），並提供兩種主要方式來與訓練過程互動：使用 YAML 配置文件的命令行介面（CLI）以及用於監控和管理任務的基於網頁的圖形使用者介面（GUI）。它包含一個實驗性管理器，可自動安裝 PyTorch、Node.js 和 FFmpeg，以使環境與系統隔離。

它適合誰使用

它專為希望使用 LoRA 和 LoKr 等技術微調擴散模型的創作者和開發者設計，特別是那些硬體資源有限或偏好簡化設置流程的人。

特色

• 廣泛模型支援：與大量圖像、影片、指令/編輯和音訊擴散模型相容。
• 自動化設置：實驗性管理器處理依賴安裝和環境建立。
• 彈性訓練：支援 LoRA 和 LoKr 訓練，並能針對特定模型層進行定位或排除。
• 簡化資料處理：自動處理圖像調整大小和長寬比，僅需圖像和匹配的文字說明檔案。
• 雲端整合：提供官方範本和指南，用於在 RunPod 和 Modal 上進行訓練。