使用的開源庫為 Stable Diffusion web UI，它是基于 Gradio 庫的 Stable Diffusion 瀏覽器界面。
運(yùn)行 Stable Diffusion 需要硬件要求比較高，運(yùn)行時(shí)會消耗較大的GPU資源，硬件環(huán)境最好是有獨(dú)立顯卡。所以我在家里的PC上進(jìn)行部署環(huán)境，此前玩WOW，買了一張獨(dú)立顯卡GTX 1660 Ti，看看能不能運(yùn)行起來。
Windows安裝git
Git安裝包下載地址：https://git-scm.com/download/win
git 安裝過程就略掉了，國內(nèi)網(wǎng)絡(luò)下載確實(shí)很慢，相信大家都會有辦法的。
下載 Stable Diffusion web UI GitHub 源碼
從github下載Stable Diffusion web UI GitHub 源碼，下載鏈接：https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
admin@DESKTOP-6EEG99D MINGW64 ~
$ git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
Cloning into 'stable-diffusion-webui'...
remote: Enumerating objects: 28998, done.
remote: Counting objects: 100% (46/46), done.
remote: Compressing objects: 100% (29/29), done.
remote: Total 28998 (delta 26), reused 28 (delta 17), pack-reused 28952
Receiving objects: 100% (28998/28998), 32.90 MiB | 2.83 MiB/s, done.
Resolving deltas: 100% (20282/20282), done.
admin@DESKTOP-6EEG99D MINGW64 ~
git安裝后，stable-diffusion-webui在路徑：C:\Users\86186\stable-diffusion-webui
如果不想安裝git，也可以下載zip格式到本地安裝
stable-diffusion模型下載
模型下載地址：https://huggingface.co/CompVis/stable-diffusion-v-1-4-original
模型大小4GB，等待下載完成……
將模型復(fù)制到stable-diffusion-webui\models\Stable-diffusion目錄：C:\Users\86186\stable-diffusion-webui\models\Stable-diffusion
同時(shí)在stable-diffusion-webui目錄下找到webui-user.bat 雙擊并運(yùn)行這個(gè)bat文件。
bat運(yùn)行成功的話，會開始自動安裝并啟動。從運(yùn)行提示來看，需要python至少是3.10.6版本。我們電腦上python的版本是3.11.5。
接著就耐心等待下載安裝torch==2.0.1 等等python文件。
安裝完成后，會自動打開本地的Web窗口：http://127.0.0.1:7860/ 基本上就算完成了。
安裝完成后，我們需要增加模型。下載后的模型仍然放到stable-diffusion-webui\models\Stable-diffusion目錄，并點(diǎn)擊刷新按鈕，就可以看到新的模型了。
在增加模型的過程，觀察GPU和內(nèi)存使用率都非常高，這點(diǎn)也對應(yīng)了前面提到的硬體資源占用問題。
嘗試進(jìn)行文字生圖時(shí)，產(chǎn)生圖片失敗，報(bào)錯(cuò)如下，按照提示和網(wǎng)上找了一些建議，一直沒有解決。
NansException: A tensor with all NaNs was produced in Unet. This could be either because there’s not enough precision to represent the picture, or because your video card does not support half type. Try setting the “Upcast cross attention layer to float32” option in Settings > Stable Diffusion or using the –no-half commandline argument to fix this. Use –disable-nan-check commandline argument to disable this check.
生成錯(cuò)誤排查處理
網(wǎng)絡(luò)上也有一些解決這個(gè)問題的提問，大家也可以參考一下：
Stable Diffusion 生成出現(xiàn)錯(cuò)誤
加載不了safetensor模型
第二天……
找到文件目錄webui-user.bat文件，并增加–no-half 參數(shù)：
@echo off
set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--no-half
call webui.bat
重新執(zhí)行bat文件：
C:\Users\86186\stable-diffusion-webui>webui-user.bat
再重新打開模型：http://127.0.0.1:7860/，輸入提示詞，簡單輸了一個(gè)古風(fēng)，俠客。發(fā)現(xiàn)已經(jīng)開始有進(jìn)圖條了，看起來終于調(diào)好了。
經(jīng)過耐心的等待，可以正常的文字生圖了，不過，跟提示詞粘了古風(fēng)的邊，俠客估計(jì)還在睡覺沒起來。
文章來源：http://www.zghlxwxcb.cn/news/detail-763155.html
重新來一次，這次的提示詞是古風(fēng)長發(fā)美女，執(zhí)行過程順便看了一下GPU，顯存100%。
再次等待后，這次正常了，來了一位古裝翩翩美少女。
以上就是運(yùn)用開源庫Stable Diffusion web UI搭建AI生圖工具的過程，耗了一些時(shí)間，不過最終有成功完成。結(jié)果還是不錯(cuò)的。文章來源地址http://www.zghlxwxcb.cn/news/detail-763155.html
推薦閱讀
- 程序猿無煩惱：讓養(yǎng)生專家來寫代碼?。?！
- 如何使用 Google 的 Gemini
- Python包管理器PIP用法大全
- 【網(wǎng)絡(luò)安全】Suspicious DNS Query（可疑的DNS查詢）
- 【Linux】Linux MTR（My TraceRoute）command
- 【Linux】在Windows WSL (Linux的Windows子系統(tǒng))上運(yùn)行的Ubuntu如何更改主機(jī)名
到了這里，關(guān)于Python開源庫Stable Diffusion web UI搭建AI生圖工具的文章就介紹完了。如果您還想了解更多內(nèi)容，請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
