Stable Diffusion LoRA訓(xùn)練Windows系統(tǒng)下啟用Lion8bit優(yōu)化器的方法
2023-05-19 17:17 作者:Impossib1e嗨 | 我要投稿
效果:

Lion8bit優(yōu)化器需要使用到facebook的bitsandbytes庫的0.38版本,但是交叉編譯的Windows版本更新較慢并且加載動態(tài)鏈接庫有BUG,故簡單修改了一下文件
在訓(xùn)練腳本根目錄下打開CMD/PWSH,啟用venv環(huán)境
? ?2.?因?yàn)榻徊婢幾g的版本并沒有cuda11.8的庫,需要重新安裝torch2.0+cu117版本
? ?3. 安裝bitsandbytes 0.38的預(yù)編譯wheel(在同一目錄下執(zhí)行)
? ?4. 使用我修改好的加載文件替換庫里自帶的(直接覆蓋venv)
? ?5. 開始使用
庫文件:https://wwpe.lanzoub.com/iqJ9t0wlgv9a?密碼:des2
bitsandbytes的wheel文件:https://wwpe.lanzoub.com/isx4N0wlhu8j?密碼:518v
P.S:這個(gè)只是臨時(shí)解決方式,因?yàn)閏udart和libbitsand的路徑直接寫死在文件里了,之后等更新吧。
標(biāo)簽: