推 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.cc), 來自: 218.173.26.142 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/C_Chat/M.1664983872.A.1ED.html
※ 本文轉寄自 ptt.cc, 文章原始頁面
Re: [閒聊] AI繪圖簡直毫無門檻
Re: 回文串
141239
[閒聊] AI繪圖簡直毫無門檻
C_Chat09/30 21:56
1322
Re: [閒聊] AI繪圖簡直毫無門檻
C_Chat10/05 15:25
511
> Re: [閒聊] AI繪圖簡直毫無門檻
C_Chat10/05 23:31
11 則留言
art1 作者的近期文章
[黑特] 國民黨請了這個教官有沒有毛骨悚然?
退贊助的廠商應該是有感到毛骨悚然吧 那些感覺不到的是要像教官被苗嘴過才會知道嗎?XDDDDDDDDDDDD 道歉後會有廠商回頭重新贊助嗎?要是沒回頭,會繼續演哭哭嗎?XDDDDDDDDDDDDD 沒犯法但觀感不佳的行為,最嚴重的就是烙跑選總
Re: [討論] 環狀線斷軌共7處
本來想說看看八卦那邊對斷軌的意見,結果居然都沒相關新聞的文章 MRT 板才有相關新聞,政黑不但有直接相關的標題,也有類似的新聞,「同樣高架遇強 震文湖線卻沒事?專家曝環狀線關鍵原因」,雖然標題沒寫,但內容有寫到王義川提出 斷軌照 搞得好像只
Re: [新聞] 彭文正又吞敗!不滿蔡英文論文案遭起訴
→ beaverz: 2019年硬塞給國圖的筆記,沒有口試紀錄, 114.44.213.178 03/31 01:26 → beaverz: 沒有指導教授簽名認證,你跟我說那是學位 114.44.213.178 03/31 01:26 →
[黑特] 原來補質詢是為了國民黨立委
國民黨立委磨刀霍霍,結果韓為了要教訓民進黨順帶把國民黨立委質詢也搞掉了 XDDDD 難怪徐會說要給補質詢,我就想說明明就是為了教訓民進黨,幹嘛還要給這群民進黨立 委補質詢 XDDDDDDDDDDDDDDDDDD 一開始猜是巨嬰抗議生效,原來
推
→
推
推
→
推
→
→
→
推
→