stable-diffusion-webui 安裝

2023-03-09 16:19 更新

Stable Diffusion是一款功能異常強大的AI圖片生成器。它不僅支持生成圖片,使用各種各樣的模型來達(dá)到你想要的效果,還能訓(xùn)練你自己的專屬模型。 

Stable Diffusion WebUI使得Stable Diffusion有了一個更直觀的用戶界面,更適合新手用戶。

注意事項

運行Stable Diffusion需要消耗較大的資源,對顯卡的要求尤其高(至少4GB的VRAM)。

如果你沒有較好的硬件資源, 以下介紹幾個可以獲取免費硬件資源的平臺:

百度飛槳(https://aistudio.baidu.com/)

Kaggle(https://www.kaggle.com/)

谷歌Colab(https://colab.research.google.com/)


Step1:安裝需要的程序

Git

首先,我們需要安裝Git。對于我們下一步的克隆來說,這是一個必不可少的工具。

Windows的安裝程序可以拷貝網(wǎng)址下載:

https://registry.npmmirror.com/-/binary/git-for-windows/v2.38.1.windows.1/Git-2.38.1-64-bit.exe

Python

然后,我們需要安裝Python 3.10。這是官方文檔上指定的版本,Windows的安裝程序可以拷貝以下網(wǎng)址下載:

https://mirrors.huaweicloud.com/python/3.10.9/python-3.10.9-amd64.exe

注意:

Windows下安裝Python時,請務(wù)必勾選“Add Python to PATH”,將Python執(zhí)行路徑添加到PATH環(huán)境變量中。

如果你選擇的是自定義安裝,請務(wù)必在安裝組件選擇中勾選PIP(PIP 是Python的包管理工具)。 


Nvidia CUDA

注意:

CUDA需要搭配Nvidia顯卡驅(qū)動進行安裝和使用。如果你使用的是AMD顯卡,則需要使用Ort。

在安裝CUDA之前,請確保你已安裝Nvidia顯卡驅(qū)動。不同的驅(qū)動版本固定搭配不同的CUDA版本。

Windows下,在這里下載安裝:https://developer.download.nvidia.cn/compute/cuda/11.7.1/local_installers/cuda_11.7.1_516.94_windows.exe


在Windows下,安裝CUDA一般只用根據(jù)安裝程序的提示就可以安裝了。如果你不想安裝太多冗余的東西,則可以只在安裝選項中勾選Runtime和Development(如果你不知道如何操作,可以在安裝完成后再在系統(tǒng)設(shè)置中卸載)。

PyTorch

PyTorch也是Stable Diffusion不可缺少的依賴包,它主要用于拓展CUDA的功能。

就像不同的Nvidia驅(qū)動版本固定搭配不同的CUDA版本一樣,不同的Torch版本也固定搭配不同的CUDA版本使用(例如: 使用PyTorch-1.13.1+CUDA11.7)。

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

如果你使用的是不同版本的CUDA,你可能需要去https://pytorch.org/get-started/locally/查看你對應(yīng)的Torch安裝版本及安裝命令(如下圖)。 


Step2:克隆Stable Diffusion+WebUI

打開Windows終端(Powershell,不是CMD)。首先,檢查磁盤的剩余空間(一個完整的Stable Diffusion大概需要占用30~40GB的剩余空間),然后進到你選好的磁盤或目錄下:

cd D:  # 此處替換成你想要克隆的地址

然后使用Git克隆AUTOMATIC1111的stable-diffusion-webui :

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git

克隆完成后,進入到克隆好的目錄。

cd .\stable-diffusion-webui

Step3:安裝和配置WebUI

配置PIP源

pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/ # 這里你也可以選擇其它的PIP源進行更換

安裝Stable Diffusion和WebUI所需的PIP包。執(zhí)行下列命令后耐心等待安裝完成。

pip install -r requirements_versions.txt # 執(zhí)行此條命令前,請檢查你的剩余磁盤空間

然后我們執(zhí)行“webui-user.bat”。

.\webui-user.bat 

根據(jù)配置優(yōu)化Stable Diffusion

直至目前,Stable Diffusion的基本框架已經(jīng)被我們安裝好了。但是根據(jù)我們配置的不同,還需要做一些簡單的設(shè)置。

在Windows下,打開webui-user.bat

將全部內(nèi)容替換為:

@echo off
set PYTHON=C:\python\python.exe
set GIT=set VENV_DIR=venv
set COMMANDLINE_ARGS=--medvram --autolaunch --deepdanbooru --xformers
call webui.bat

注:此處的“--medvram”是針對6GB及以上顯存的顯卡優(yōu)化的,根據(jù)顯卡配置的不同,你還可以更改為“--lowvram”(4GB以上)、“--lowram”(16GB以上)或者刪除此項(無優(yōu)化)。此外,此處的“--xformers”選項可以開啟Xformers。加上此選項后,顯卡的VRAM占用率就會相較原來減少一半,能大幅提升圖片生成效率。

Step4:下載Stable Diffusion模型

模型相當(dāng)于Stable Diffusion的核心。在沒有模型的情況下,Stable Diffusion是無法啟動的。

所以,我們需先下載模型。Hugging Face(https://huggingface.co)是學(xué)習(xí)使用Stable Diffusion的不可或缺的平臺。

你可以從此網(wǎng)站的Model版塊中搜索你需要的模型名稱來下載模型。

Stable Diffusion的模型后綴名一般為ckpt。偶爾,模型也會附帶VAE(.vae.pt)或配置文件(.yaml)。我們需要把ckpt模型、VAE以及配置文件放在models目錄下的Stable-diffusion目錄中。

注意:如果一個模型附帶配置文件或者VAE,你則需要先把它們的文件名改為相同的文件名,然后再放入目錄中,否則這個模型的配置可能無法正確讀取,影響圖片生成效果。

Step5:配置你的Stable Diffusion WebUI

模型配置完成后,重新回到克隆目錄中。

cd D:\stable-diffusion-webui

運行webui-user.bat。

.\webui-user.bat

如果你的終端顯示的內(nèi)容與下圖相符,那么恭喜你,Stable Diffusion WebUI的基本框架安裝成功了!


在瀏覽器打開頁面:



以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號