用免費GPU部署自己的stable-diffusion專案(AI生成圖片)

麻雀小妖發表於2024-03-21

2021年時出現了 openAI 的 DALL,但是不開源。2022年一開年,DALL-E 2釋出,依然不開源。同年7月,Google 公佈其 Text-to-Image 模型 Imagen,並且幾乎在同一時間段AI影像生成平臺 Midjourney 也進行公測。同年8月,Stable Diffusion 的釋出將AIGC和AI繪畫徹底帶出了圈。
Stable Diffusion 是屬於生成模型(Generative Model)的,現在主流的AI繪畫模型就是各種 finetune 後的 Stable Diffusion。

關於Stable Diffusion的介紹可以參考:https://blog.csdn.net/starzhou/article/details/136557787

本文介紹的是一種能夠方便快捷地部署一個stable-diffusion專案的方法,適合初學者。

Step1.註冊賬號

1、註冊連結:https://growthdata.virtaicloud.com/t/SA,註冊後會獲得168算力,足夠試用了。
2、註冊後進入平臺中心:https://platform.virtaicloud.com/

Step2.建立專案

1、選擇:當前空間,需要確保當前所在空間是註冊時系統自動生成的空間,否則沒有算力。
2、點選:快速建立,選擇建立專案,建立新專案。

3、填寫相關的專案資訊,其中映象選擇如下:

4、資料集選擇如下:

5、點選“建立”按鈕。

6、彈出的上傳程式碼對話方塊,選擇 "暫不上傳"。

Step3. 初始化開發環境例項

1、點選左上角的“執行程式碼”

2、進行資源配置。以下是我嘗試的可選擇的最大配置。

如果要選擇更大的配置,有時會提示需要選擇版本更高的映象,有時會提示“配額不足,如需擴容請聯絡客服”。

目前還沒去試一下還有其他哪些映象可用,也沒試過配額不足的話能不能使用,如有成功的小夥伴歡迎留言分享經驗。

SSH選擇“開啟”。如果還未開啟過,旁邊會有個提示,可以參照文件進行設定。

開發者工具選擇JupyterLab。

3、配置開放埠,埠使用目的是必填的。

4、一定要記得設定最長執行時間,否則如果忘了停止專案,一直執行會浪費算力。

5、執行初始化中,需要耐心等待。

6、當右上角的狀態變為“執行中”,右下角的JupyterLab網頁終端 不再是灰色時,表明工具注入成功。

點選“進入開發環境”,或者右下角的“網頁終端”,開始執行程式碼。

Step4. 部署模型

在下面的視窗中依次執行如下程式碼(複製貼上過去):

1、解壓程式碼及模型

tar xf /gemini/data-1/stable-diffusion-webui.tar -C /gemini/code/
等待出現新的一行可輸入命令的游標行,說明上一條命令執行完成。

2、複製frpc內網穿透檔案

chmod +x /root/miniconda3/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2

3、複製模型檔案到專案目錄下

cp /gemini/data-1/v1-5-pruned-emaonly.safetensors /gemini/code/stable-diffusion-webui/

4、更新系統httpx依賴

pip install httpx==0.24.1
這個命令刷屏比較多,需要耐心等一下,完成後是下面這樣。

5、執行專案

cd /gemini/code/stable-diffusion-webui && python launch.py --deepdanbooru --share --xformers --listen
當命令視窗出現如下兩個網址時表示部署成功,將右側邊欄往下滑,找到“埠資訊”,將外部訪問連結複製到瀏覽器開啟。

開啟網址可能需要一些時間,請耐心等待。

輸入提示語(可以從網上找一些),點選“生成”按鈕。有時圖片一直載入不出來,可換個時間段再試。

提示語示例:

[Description of the character], young woman, balck hair, smiling eyes, graceful figure, dynamic pose, gentle breeze, cityscape background, golden light, alluring charm, highly detailed, realistic, cinematic, unreal engine, 16K

生成圖片的效果示例:

6、停止專案

儲存專案:將當前環境製作為映象,然後點選“停止”,以免浪費算力。

整個兒專案部署過程非常簡單,也就大概半個小時,有時間可以多做些其他嘗試。

相關文章