網友評分: 5分
Cybertron Furnace是一款lora模型訓練軟件,準備好訓練的圖片,尺寸根據你顯卡顯存的大小,默認的圖片尺寸大小為512×512,顯卡顯存小于8G的,建議使默認尺寸,如果你的顯卡顯存可以滿足12g或12g以上,推薦使用768×768的圖片來訓練我們的模型,用大尺寸訓練后的模型可以適當減少生成寬圖出現的肢體疊加等效果。
一、圖片要求
至少15張圖片,每張圖片的訓練步數不少于100
照片人像要求多角度,特別是臉部特寫(盡量高分辨率),多角度,多表情,不同燈光效果,不同姿勢等
圖片構圖盡量簡單,避免復雜的其他因素干擾
可以單張臉部特寫+單張服裝按比例組成的一組照片(這里比例是3:1)
減少重復或高度相似的圖片,避免造成過擬合
二、圖片處理
裁剪你的照片成768×768(我這里是用了768×768)的比例,裁剪照片可以到birme站點裁剪后批量下載,或者利用picpick軟件自己裁剪后期用python腳本批量改名和批量二次裁剪。
準備圖片解析詞,用stable diffusion 圖片預處理模塊preprocess image進行圖片解析,注意圖片尺寸修改為我們裁剪后的圖片尺寸,圖片解析使用BLIP,圖片少于15張建議勾選create Flipped copies選項。
(可選)編輯我們生成好的解析詞文件,加入我們的關鍵人物tag,如果是服飾圖片,可以給服飾加上我們自定義的tag用于區分服飾,相同的發型也可以打上發型的自定義tag,后面使用該lora模型可以加上服飾或發型部分的tag用于生成對應要求的圖像,這里圖片解析詞我們可以使用kohay_ss的Utilities下的Captioning批量給我們處理后的解析詞文件增加對應的角色tag和服飾tag,或者利用軟件
批量增加我們的tag,推薦使用BooruDatasetTagManager,可以自己選中多張圖加入我們要的tag,人物我們可以增量加入常用的tag如face,nose,lips,hairstyle,eyes,ears,forehead,breast等。
三、開始訓練
準備好kohay_ss GUI訓練工具(這里使用拓展的訓練工具)并成功啟動
建立訓練所需的三個文件夾image,model,log三個文件夾,然后在image目錄下新建一個文件夾,文件夾命名符合以下規則形(數字_角色關鍵詞)136_zoogygh
數字部分?。?500除以圖片數量和100的最大值),每張圖片訓練步數起步最低100
加載基礎的訓練配置文件,這里網上有整理了針對基礎配置和低顯存配置的兩個默認的訓練配置文件,我們加載進來后,默認基于stable diffusion1.5的模型訓練的,這里看個人需求,可以選擇custom后加載自己的模型文件
把我們預處理后的圖片拷貝到數字_關鍵詞的目錄下面,然后開始設置我們剛剛定義好的三個文件夾目錄位置,注意image文件目錄不需要選擇到我們的圖片,直接到image這層路徑就好了
微調我們的訓練參數
開始訓練
四、模型使用
1、訓練完后會在我們剛剛定義的model目錄下生成一個lora模型文件,我們將他拷貝到stable diffusion的model目錄下的lora文件夾下
2、安裝lora加載插件,安裝后可以在圖生圖的快捷區域調出我們的模型文件,復制一段提詞后,點擊我們的lora模型,在正面提詞里修改我們的lora模型權重,或者加上服飾tag,開始抽卡
下載后,解壓縮在除C盤以外的硬盤中(實在是只有C盤也可以試下),打開“cfurnace_ui”文件夾,雙擊“Cybertron Furnace.exe“即可開啟煉丹之旅。首次啟動賽博丹爐,需要等待控制臺加載一些內容,看到server start,即可點擊“開啟煉丹爐”。同時,首次煉丹的用戶(以前電腦上沒有煉過丹),會需要下載1.7G左右的訓練用文件,需要等待一段時間,之后再打開就不需要了。
希望這個工具可以幫助新手朋友們快速成長,不再為工具安裝、使用報錯而煩惱。
煉出更多的好丹,創作更多的AI作品,享受AIGC時代
標簽: lora模型訓練