使用LoraTrainer進(jìn)行Lora模型訓(xùn)練

要求
支持Cuda的Nvidia的顯卡,不支持AMD等其他的顯卡
使用一鍵包之前,請(qǐng)確保cuda正確的安裝
其他的軟件已經(jīng)完全的包含在了包中,不需要額外的下載,點(diǎn)開(kāi)即用


創(chuàng)建一個(gè)工程
點(diǎn)擊 文件→新建項(xiàng)目

填寫(xiě)項(xiàng)目名稱和路徑。生成的路徑為 路徑+名稱
空的項(xiàng)目創(chuàng)建成功。

處理訓(xùn)練集
訓(xùn)練集可以通過(guò)自動(dòng)導(dǎo)入的方式導(dǎo)入,也可以通過(guò)手動(dòng)的方式導(dǎo)入。
自動(dòng)導(dǎo)入
點(diǎn)擊 源文件→自動(dòng)導(dǎo)入

在菜單中選擇導(dǎo)入方式,導(dǎo)入圖片。結(jié)果如下

如果是二次元圖片建議選擇Deepdanbooru的打標(biāo)器,如果是其他的圖片請(qǐng)選擇Clip,也可以不選,則不會(huì)給圖片添加任何標(biāo)簽。
點(diǎn)擊導(dǎo)入,等待處理完成。如果出現(xiàn)長(zhǎng)時(shí)間卡住等情況,請(qǐng)嘗試中斷后再導(dǎo)入。

完成后就處理完畢了,如果自動(dòng)導(dǎo)入處理的圖片不完美,請(qǐng)看下一節(jié),沒(méi)有問(wèn)題則直接跳到訓(xùn)練。

手動(dòng)處理
刪除掉不完美的素材
接上文,這里需要?jiǎng)h除掉一些不完美的素材。使用ctrl或者shift可以選中圖片,這里選中之后,在菜單中選擇刪除選中的預(yù)處理圖片。

重新裁剪圖片
切換到原圖的頁(yè)面,進(jìn)行處理。
注:這里可以將預(yù)處理文件過(guò)濾器切換到不包含,就可以顯示沒(méi)有被與預(yù)處理過(guò)的圖片。

點(diǎn)擊圖片,再點(diǎn)擊裁剪圖片

拖動(dòng)框選裁剪框,然后保存

完成想要的圖片裁切后,切回到數(shù)據(jù)集頁(yè)面,就可以看到完成后的數(shù)據(jù)集。

重新打標(biāo)
裁剪后需要對(duì)圖片進(jìn)行重新打標(biāo),這里使用自動(dòng)打標(biāo)。
過(guò)濾器中可以選擇只顯示無(wú)標(biāo)簽的預(yù)處理文件。

選擇后,點(diǎn)擊菜單 識(shí)別選中的圖片

根據(jù)自己的圖片選擇打標(biāo)器的類型,閾值可以更改識(shí)別準(zhǔn)確度,過(guò)濾掉低可能性的標(biāo)簽,點(diǎn)擊識(shí)別。
識(shí)別完成后,可以點(diǎn)擊每個(gè)圖片選擇是否使用這個(gè)標(biāo)簽。后面的數(shù)字代表標(biāo)簽的可能性。

點(diǎn)擊保存,回到數(shù)據(jù)集頁(yè)面,切換過(guò)濾器為所有。
模型訓(xùn)練
設(shè)置觸發(fā)詞
點(diǎn)擊上方 標(biāo)簽→ 添加標(biāo)簽。
注:會(huì)將標(biāo)簽添加至當(dāng)前顯示的預(yù)處理圖片中。如果想給所有的圖片添加,請(qǐng)?jiān)跀?shù)據(jù)集文件夾內(nèi)點(diǎn)擊”所有”
英文逗號(hào)分隔標(biāo)簽

創(chuàng)建訓(xùn)練集文件夾
將所有的圖片創(chuàng)建至一個(gè)新的文件夾內(nèi),也可以根據(jù)自己的需要?jiǎng)?chuàng)建。
全選后點(diǎn)擊使用選中的圖片創(chuàng)建新的文件夾。
也可以使用上方的菜單中的“數(shù)據(jù)集”新建數(shù)據(jù)集“的方式創(chuàng)建空的數(shù)據(jù)集。使用復(fù)制、粘貼、刪除、剪切功能,可以像文件瀏覽器一樣處理。

填入數(shù)據(jù)集信息,名字根據(jù)自己需要,步數(shù)填寫(xiě)100,如果有自己需求,按照自己需求填寫(xiě)

配置訓(xùn)練參數(shù)
選項(xiàng)卡切換至訓(xùn)練配置
軟件已經(jīng)內(nèi)置了訓(xùn)練配置,不需要過(guò)多的設(shè)置。

填寫(xiě)好模型名稱
如果是完整版的軟件,可以選擇預(yù)訓(xùn)練模型庫(kù)的v2選項(xiàng)。這樣就不需要額外地從網(wǎng)絡(luò)上下載很大的模型。
點(diǎn)擊保存。
點(diǎn)擊啟動(dòng)。

啟動(dòng)后就可以看到訓(xùn)練腳本正在運(yùn)行。

面板切換到監(jiān)視器能看到進(jìn)度,等待完成。

完成后可以打開(kāi)輸出的文件夾

模型評(píng)測(cè)
這里主要針對(duì)完成后用模型生成圖片,你也可以不用這個(gè)功能,直接將模型導(dǎo)入到stable-duffusion-webui內(nèi)自己使用。
選項(xiàng)切換到模型頁(yè)面,選擇一個(gè)模型。

啟動(dòng)預(yù)覽服務(wù)

服務(wù)會(huì)啟動(dòng)一個(gè)stable-diffusion-webui

完成后會(huì)看到狀態(tài)的改變。關(guān)閉服務(wù)時(shí),直接將彈出的窗口關(guān)掉即可。

可以在設(shè)置 -> 預(yù)覽服務(wù)配置里面來(lái)設(shè)置啟動(dòng)的參數(shù)。
注:也可以從其他的路徑來(lái)加載模型。路徑一般為:
<我的stable-diffusion-webui>\models\Stable-diffusion

點(diǎn)擊生成預(yù)覽圖,填寫(xiě)如入?yún)?shù),點(diǎn)擊應(yīng)用即可,這里加入了剛才訓(xùn)練前加入的觸發(fā)詞,可以有效的生成想要的圖片。
需要輸入使用的模型,和lora的比值。后者決定了模型在生成過(guò)程的權(quán)重。

這樣就可以生成模型的圖片了。


關(guān)于
軟件正在處于最早期的開(kāi)發(fā)階段,可能存在一定量的bug,本次放出主要是為了讓更多的人進(jìn)行測(cè)試,提出改進(jìn)意見(jiàn)和驗(yàn)證程序功能。
如果發(fā)現(xiàn)任何問(wèn)題或者有好的建議,可以直接在下方評(píng)論區(qū)提出。
下載地址
完整版下載地址:
鏈接:https://pan.baidu.com/s/1zyY2WZs_OWR8AGtYQnYnPQ?pwd=3hmz?
提取碼:3hmz?