PTT推薦

Re: [閒聊] AI繪圖簡直毫無門檻

看板C_Chat標題Re: [閒聊] AI繪圖簡直毫無門檻作者
art1
(人,原來不是人)
時間推噓 5 推:5 噓:0 →:6

conqueror507: 我也是遇到Torch is not able to use GPU 10/05 15:53

會有這個訊息是因為 launch.py 中有一段程式碼在檢查是否有 gpu 可用,當沒有 gpu 可用時,就會跳這個訊息出來,同時也會提示要去修改命令列傳入的參數,才能略過這 段檢查,也就是 webui-user.sh 檔案中的這一行 export COMMANDLINE_ARGS=""

把提示的參數填入成對的雙引號中,像是這樣
export COMMANDLINE_ARGS="--skip-torch-cuda-test"

嘗試了用 token 下載 huggingface 的檔案,速度超級快,上百 M 在跑,因此就改用 這方法下載模型參數了,畢竟 colab 的 GPU 時間可是很寶貴的,不能浪費在下載檔案 上,申請 huggingface 的帳號再取得 access token 用不了多少時間

同時還研究了原始碼,發現 --ckpt-dir 的參數優先級比較高,因此下載回來的模型參 數檔案 ckpt 也不需要設定軟連結或是移動到預設的資料夾了,只需要把 --ckpt-dir 的參數加進 COMMANDLINE_ARGS 中,就能下載一堆模型參數檔回來,到時再從 web 介面 的 Settings 分頁的 Stable Diffussion 下拉選單中設定要使用哪個 ckpt 檔就好

ckpt 檔一多,還多一個融合兩個 ckpt 檔的玩法,在 web 介面的 Checkpoint Merge 分頁可以設定,只是目前我下載的兩個 ckpt 檔,一個是瘦身過的,不知道能不能正常 融合

目前有用 GPU 的話,COMMANDLINE_ARGS 長這樣
COMMANDLINE_ARGS="--share --skip-torch-cuda-test --ckpt-dir /content"
/content 是 colab 預設的資料夾位置

使用 CPU 的話,則要改成
COMMANDLINE_ARGS=
"--share --precision full --no-half --skip-torch-cuda-test --ckpt-dir /content"

不過 colab 上都是用指令改了,所以實際內容還會有一點差異,上面換行只是為了顯示 方便

不想花錢試玩的就用 colab 玩玩吧,google 帳號越多,每天能玩的時間就越多

--
「看在上帝的份上,我們不能當著孩子的面槍殺他的父親!」
「那我們就連孩子一起幹掉!」
「第一槍打中的是那個天真的孩子,第二槍是抱著孩子的母親。第三槍
停頓了三秒,當難以致信的父親擦拭著臉上愛子的腦漿從震驚中醒轉
過來,慢慢的轉頭看向停車的地方,分辨出是誰在攻擊他後,第三顆
子彈才將他的腦殼打飛。」 ─摘自〈狼群〉

--

※ PTT留言評論
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 218.173.26.142 (臺灣)
PTT 網址

t7713356200310/05 23:36毫無門檻.........

tym748210/05 23:38嗯嗯跟我想的差不多

orze0410/05 23:42colab 有現成的腳本可以用

orze0410/05 23:53本地機器我是這樣設定

orze0410/05 23:53https://i.imgur.com/WnOWtU7.png

圖https://i.imgur.com/WnOWtU7.png?e=1666681277&s=WTOnmugbbL5olxNmTIzyKg, AI繪圖簡直毫無門檻

orze0410/05 23:58https://i.imgur.com/CzjA6hV.png 這張才對

圖https://i.imgur.com/CzjA6hV.png?e=1666653256&s=U9qS5spaPxVZsB3HcTd3jw, AI繪圖簡直毫無門檻

art110/05 23:59merge 失敗了,看來還要繼續研究程式碼看為何會這樣...

art110/06 01:59看來是記憶體不夠,就會導致行程被砍掉

hitlerx10/06 04:58毫無門檻.....

TaiwanXDman10/06 05:41你這內文跟標題完全不符啊XDD

art110/07 21:09用 colab 玩的門檻很低了吧 XDD