使用云來(lái)做Dreambooth, Lora模型訓(xùn)練詳細(xì)教程
治障君在制作灌籃高手真人化這期視頻的時(shí)候發(fā)現(xiàn)了攬睿AI這家新的云顯卡平臺(tái)。
在制作視頻的過(guò)程中，治障君也把自己訓(xùn)練用的kohya_ss和常用模型做成了鏡像和數(shù)據(jù)集，方便其他有需要的觀眾使用。以下是如何使用治障君發(fā)布的鏡像來(lái)進(jìn)行你自己的訓(xùn)練：
首先，登錄攬睿，注冊(cè)一個(gè)你自己的賬號(hào)
https://www.lanrui-ai.com/register
攬?，F(xiàn)在推廣期間，如果你使用治障君提供的邀請(qǐng)碼注冊(cè) 3269，你將獲得攬睿的新人免費(fèi)訓(xùn)練禮包一份。
2. 來(lái)到工作空間，然后點(diǎn)擊右上角的創(chuàng)建工作空間
3. 選擇一臺(tái)云主機(jī)，特價(jià)的3090應(yīng)該是1.9一個(gè)小時(shí)，如果圖片不多，只是跑幾個(gè)epoch的訓(xùn)練，3090一個(gè)小時(shí)應(yīng)該綽綽有余。
4. 工作空間的名字可以取自己容易辨認(rèn)的，運(yùn)營(yíng)環(huán)境鏡像在公有鏡像里面選擇，kohya_ss，掛載網(wǎng)盤，數(shù)據(jù)盤可以選擇不使用，數(shù)據(jù)集選擇zzj-sd-models，里面有治障君推薦的目前常用的Stable Diffusion Checkpoint模型，以及Lora和embedding。完成之后點(diǎn)擊創(chuàng)建實(shí)例。如果你想要找更多的模型，也可以在c站www.civitai.com或者煉丹閣www.liandange.com上面下載再上傳到自己的文件夾里面。
5. 創(chuàng)建完成后，實(shí)例會(huì)進(jìn)入啟動(dòng)開機(jī)狀態(tài)，這里要稍等片刻（2-5分鐘左右），等待服務(wù)器啟動(dòng)。
顯示運(yùn)行中的時(shí)候，就可以在操作這里點(diǎn)擊進(jìn)入
6. 進(jìn)入Jupyter之后，data文件夾是你自己的網(wǎng)盤，你的訓(xùn)練數(shù)據(jù)可以全部存放在這里，長(zhǎng)期有效，在你的賬戶中不會(huì)丟失。imported_datasets是治障君發(fā)布的數(shù)據(jù)集，里面你可以找到各種常用的model，imported_models里面也有一些官方放入的常用模型。點(diǎn)擊Terminal開啟命令去。
7. 按照一下步驟依次鍵入來(lái)啟動(dòng)kohya_ss：
cd /app/kohya_ss
source venv/bin/activate
accelerate config
看到這句話出現(xiàn)后，Please select a choice using the arrow or number keys, and selecting with enter，接下去會(huì)出現(xiàn)以下選項(xiàng)，按照如下依次填寫
問(wèn)題1：選This machine
問(wèn)題2：選No distributed training
問(wèn)題3：輸入no
問(wèn)題4：輸入no
問(wèn)題5：輸入no
問(wèn)題6：輸入all
問(wèn)題7：選fp16
選完之后，繼續(xù)復(fù)制命令
python kohya_gui.py --server_port 27777 --listen 0.0.0.0
這句輸入完成之后，看到下面這段話就說(shuō)明kohya就啟動(dòng)成功? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ??
Running on local URL:? http://0.0.0.0:27777
To create a public link, set `share=True` in `launch()`.
8.回到剛才的工作空間界面，點(diǎn)擊實(shí)例名稱
9. 進(jìn)入之后，點(diǎn)擊復(fù)制調(diào)試地址
10. 在游覽器中粘貼剛剛的地址，回車就能看到ui的畫面
11. 以上基本就是kohya的啟動(dòng)方式，治障君在data下面設(shè)置了一個(gè)叫training_data的文件夾，里面做的是如下設(shè)置，你也可以用來(lái)參考：
12. 訓(xùn)練的模型路徑可以參考以下：
對(duì)應(yīng)的是你的模型在數(shù)據(jù)集下面的鏈接，治障君這里用的是anything，你可以把a(bǔ)nything的模型名字換成你自己想要的底模的文件名：
/ark-contexts/imported_datasets/zzj-sd-models/Stable-diffusion/Anything_V3.ckpt
13. 如果你剛才和治障君使用的是一樣的文件夾格式，那你可以直接輸入以下的目錄來(lái)配置你的訓(xùn)練:
/ark-contexts/data/training_data/imgs
/ark-contexts/data/training_data/model
/ark-contexts/data/training_data/log
14. 打標(biāo)的話，圖片需要先上傳到img下面的文件夾。如果你剛開始接觸訓(xùn)練的話，可以參考治障君的這期視頻來(lái)了解訓(xùn)練的基本設(shè)置。
看Lora怎么樣訓(xùn)練模型，AI真是把他的樣子學(xué)到骨子里了！
15. 最后，填完你的設(shè)置內(nèi)容，直接點(diǎn)Train model，你就可以開始你的煉丹之旅啦。Dreambooth的訓(xùn)練方法也可以參照這個(gè)教程。
16. 你的訓(xùn)練進(jìn)度可以在Jupter的窗口里面查看
17. 最后，煉完的丹在你training_data/model的目錄，直接下載取丹即可。enjoy~
