練自己的AI圖像LoRA模型)
kohya_ss10分鐘上手零代碼訓(xùn)練自己的AI圖像LoRA模型【免費(fèi)下載鏈接】kohya_ss項(xiàng)目地址: https://gitcode.com/GitHub_Trending/ko/kohya_sskohya_ss 是一個(gè)開(kāi)源的AI圖像模型訓(xùn)練工具它用圖形界面把命令行訓(xùn)練流程封裝成填表式的操作。你不需要會(huì)寫(xiě)代碼只要有一塊NVIDIA顯卡就能訓(xùn)練LoRA、Dreambooth或微調(diào)Stable Diffusion系列模型。它適合想讓自己的AI圖像模型學(xué)會(huì)特定人物、畫(huà)風(fēng)或產(chǎn)品的個(gè)人創(chuàng)作者和進(jìn)階玩家。kohya_ss 是什么圖形界面 命令行訓(xùn)練一體它是 Kohya 的 Stable Diffusion 訓(xùn)練腳本的圖形界面版本你在界面上勾選參數(shù)它就自動(dòng)幫你生成并執(zhí)行訓(xùn)練命令出錯(cuò)時(shí)能直接看到命令行內(nèi)容方便排查支持 LoRA、LoHa、LoKr、Dreambooth、全量微調(diào)和 Textual Inversion 多種訓(xùn)練方式覆蓋 SD1.5/2.x、SDXL、SD3、Flux.1、Lumina Image 2.0、Anima、HunyuanImage-2.1 等主流底模除了本地訓(xùn)練也支持 Colab、Runpod、Novita、Docker 等云端方案沒(méi)有顯卡的人可以用瀏覽器跑環(huán)境準(zhǔn)備安裝前需要哪些硬件和依賴硬件方面NVIDIA 顯卡訓(xùn)練 SDXL 的 LoRA 建議至少 12GB 顯存SD1.5 的 LoRA 對(duì)顯存要求低得多能訪問(wèn) Python 3.10 或 3.11.9、Git 的電腦Windows、Linux、macOS 均可依賴方面項(xiàng)目自帶一鍵腳本會(huì)自動(dòng)裝好 PyTorch、diffusers、gradio 等全部依賴你只需要git clone --recursive https://gitcode.com/GitHub_Trending/ko/kohya_ss cd kohya_ss然后啟動(dòng)圖形界面Windows 雙擊或運(yùn)行g(shù)ui-uv.batLinux/macOS 運(yùn)行./gui-uv.sh# Windows .\gui-uv.bat # Linux / macOS ./gui-uv.sh腳本第一次運(yùn)行會(huì)自動(dòng)安裝 Python 環(huán)境和依賴耐心等它跑完。瀏覽器打開(kāi)后你看到的就是訓(xùn)練界面。各平臺(tái)的詳細(xì)步驟可以對(duì)照 docs/Installation/uv_windows.md 這份安裝說(shuō)明。最小可用流程從一批照片到一個(gè)LoRA文件這里有個(gè)坑先說(shuō)清楚訓(xùn)練前要把數(shù)據(jù)集整理好。把要學(xué)習(xí)的圖片jpg、png 均可放進(jìn)一個(gè)文件夾每張圖配一個(gè)同名 txt 文件寫(xiě)圖片描述比如1girl, white shirt。接下來(lái)按這個(gè)順序操作輸入打開(kāi) LoRA 標(biāo)簽頁(yè)指定底模文件如 SD1.5 或 SDXL 模型和上面的數(shù)據(jù)集文件夾關(guān)鍵操作設(shè)置分辨率、學(xué)習(xí)率等參數(shù)界面里都有默認(rèn)值點(diǎn)開(kāi)始訓(xùn)練。訓(xùn)練過(guò)程中可以生成樣本圖隨時(shí)查看模型學(xué)到的效果對(duì)不對(duì)得到什么訓(xùn)練結(jié)束后在輸出目錄得到一個(gè).safetensors格式的 LoRA 文件。把它放進(jìn) WebUI 或 ComfyUI 的 LoRA 目錄出圖時(shí)加載就能生成你訓(xùn)練的主題了LoRA 各參數(shù)的含義和推薦值建議直接看項(xiàng)目?jī)?nèi)置的 docs/LoRA/top_level.md里面有 SDXL LoRA 的完整調(diào)參建議。按顯存選方案不同顯卡怎么選訓(xùn)練方式你的硬件條件推薦方案說(shuō)明8GB 顯存如 RTX 3060SD1.5 LoRA 訓(xùn)練數(shù)據(jù)集控制在幾十張圖分辨率 512可穩(wěn)定跑完12GB 顯存如 RTX 3060 12GSDXL LoRA 訓(xùn)練官方文檔給出的 SDXL LoRA 最低顯存門(mén)檻建議同時(shí)勾選只訓(xùn)練 UNet 選項(xiàng)16GB 以上顯存SDXL LoRA / 全量微調(diào)可以放大分辨率和 batch size訓(xùn)練速度明顯提升無(wú)獨(dú)立顯卡Colab / Runpod 云端用瀏覽器完成全部流程本地只做結(jié)果下載不確定自己屬于哪檔時(shí)先用 SD1.5 跑通一次完整流程再上 SDXL問(wèn)題最好排查。常見(jiàn)問(wèn)題三個(gè)高頻報(bào)錯(cuò)的解決思路現(xiàn)象啟動(dòng)時(shí)報(bào) No module called tkinter。原因當(dāng)前 Python 版本缺少 tkinter 組件常見(jiàn)于自行安裝的 Python 3.10。 怎么辦重新安裝一遍 Python 3.10安裝時(shí)勾選 tcl/tk 組件或用uv方式啟動(dòng)讓腳本管理獨(dú)立環(huán)境?,F(xiàn)象遠(yuǎn)程服務(wù)器SSH/云主機(jī)上點(diǎn)開(kāi)始后流程卡住不動(dòng)。原因沒(méi)加--headless參數(shù)界面的本地文件選擇和覆蓋確認(rèn)彈窗跑在了無(wú)顯示的服務(wù)器上永遠(yuǎn)等不到你的點(diǎn)擊。 怎么辦用./gui-uv.sh --headless啟動(dòng)它會(huì)自動(dòng)隱藏本地文件選擇框并在重名時(shí)直接覆蓋。現(xiàn)象Windows 上訓(xùn)練報(bào)錯(cuò)提到 page file頁(yè)面文件。原因系統(tǒng)虛擬內(nèi)存上限不夠訓(xùn)練進(jìn)程申請(qǐng)內(nèi)存時(shí)被拒絕。 怎么辦在 Windows 設(shè)置里把頁(yè)面文件改大或給訓(xùn)練留出更多系統(tǒng)內(nèi)存余量。適用場(chǎng)景想讓自己喜歡的角色或?qū)櫸镌谒谐鰣D里保持一致訓(xùn)練人物 LoRA給商拍產(chǎn)品生成多角度場(chǎng)景圖訓(xùn)練產(chǎn)品/風(fēng)格 LoRA把某個(gè)畫(huà)師的風(fēng)格遷移到自己的出圖流程里在云端 GPU 上掛機(jī)訓(xùn)練本地電腦只做結(jié)果預(yù)覽結(jié)尾建議建議先拿 10~20 張同主題圖片跑一個(gè) SD1.5 LoRA確認(rèn)樣本圖效果符合預(yù)期后再加數(shù)據(jù)、上 SDXL調(diào)參思路可以對(duì)著 docs/LoRA/top_level.md 逐項(xiàng)核對(duì)這是整個(gè)倉(cāng)庫(kù)里對(duì)新手最友好的一份資料。【免費(fèi)下載鏈接】kohya_ss項(xiàng)目地址: https://gitcode.com/GitHub_Trending/ko/kohya_ss創(chuàng)作聲明:本文部分內(nèi)容由AI輔助生成(AIGC),僅供參考