42 lines
1.5 KiB
PowerShell
42 lines
1.5 KiB
PowerShell
$Env:HF_HOME = "huggingface"
|
|
function InstallFail {
|
|
Write-Output "安装失败。"
|
|
Read-Host | Out-Null ;
|
|
Exit
|
|
}
|
|
|
|
function Check {
|
|
param (
|
|
$ErrorInfo
|
|
)
|
|
if (!($?)) {
|
|
Write-Output $ErrorInfo
|
|
InstallFail
|
|
}
|
|
}
|
|
|
|
if (!(Test-Path -Path "venv")) {
|
|
Write-Output "正在创建虚拟环境..."
|
|
python -m venv venv
|
|
Check "创建虚拟环境失败，请检查 python 是否安装完毕以及 python 版本。"
|
|
}
|
|
|
|
.\venv\Scripts\activate
|
|
Check "激活虚拟环境失败。"
|
|
|
|
Write-Output "安装程序所需依赖 (已进行国内加速，若无法使用加速源请用 install.ps1)..."
|
|
Set-Location .\sd-scripts
|
|
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 -f https://mirror.sjtu.edu.cn/pytorch-wheels/torch_stable.html -i https://mirrors.bfsu.edu.cn/pypi/web/simple
|
|
Check "torch 安装失败，请删除 venv 文件夹后重新运行。"
|
|
pip install --upgrade -r requirements.txt -i https://mirrors.bfsu.edu.cn/pypi/web/simple
|
|
Check "其他依赖安装失败。"
|
|
pip install -U -I --no-deps https://jihulab.com/api/v4/projects/82097/packages/pypi/files/e8508fe14c8f2552a822f5e6f5620b24fdd4ba3129c2a31a39b56425bcc023bc/xformers-0.0.14.dev0+torch12-cp310-cp310-win_amd64.whl
|
|
Check "xformers 安装失败。"
|
|
|
|
Write-Output "安装 bitsandbytes..."
|
|
cp .\bitsandbytes_windows\*.dll ..\venv\Lib\site-packages\bitsandbytes\
|
|
cp .\bitsandbytes_windows\cextension.py ..\venv\Lib\site-packages\bitsandbytes\cextension.py
|
|
cp .\bitsandbytes_windows\main.py ..\venv\Lib\site-packages\bitsandbytes\cuda_setup\main.py
|
|
|
|
Write-Output "安装完毕。"
|
|
Read-Host | Out-Null ; |