CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

deps

Browse Source
This commit is contained in:
Haoming 2026-04-13 14:06:23 +08:00
parent 603d9dea68
commit 6fe678adc5
2 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
modules/launch_utils.py
View File

@ -246,8 +246,6 @@ def run_extensions_installers(settings_file):
run_extension_installer(path)
startup_timer.record(dirname_extension)
return
re_requirement = re.compile(r"\s*(\S+)\s*==\s*([^\s;]+)\s*")
@ -289,9 +287,9 @@ def requirements_met(requirements_file):
def prepare_environment():
torch_index_url = os.environ.get("TORCH_INDEX_URL", "https://download.pytorch.org/whl/cu130")
torch_command = os.environ.get("TORCH_COMMAND", f"pip install torch==2.10.0+cu130 torchvision==0.25.0+cu130 --extra-index-url {torch_index_url}")
xformers_package = os.environ.get("XFORMERS_PACKAGE", f"xformers==0.0.34 --extra-index-url {torch_index_url}")
bnb_package = os.environ.get("BNB_PACKAGE", "bitsandbytes==0.49.1")
torch_command = os.environ.get("TORCH_COMMAND", f"pip install torch==2.11.0+cu130 torchvision==0.26.0+cu130 --extra-index-url {torch_index_url}")
xformers_package = os.environ.get("XFORMERS_PACKAGE", f"xformers==0.0.35 --extra-index-url {torch_index_url}")
bnb_package = os.environ.get("BNB_PACKAGE", "bitsandbytes==0.49.2")
packaging_package = os.environ.get("PACKAGING_PACKAGE", "packaging==26.0")
gradio_package = os.environ.get("GRADIO_PACKAGE", "gradio==4.40.0 gradio_rangeslider==0.0.8")
@ -347,16 +345,16 @@ assert cuda or xpu or mps
v_CUDA = f"{ver_CUDA[0:-1]}.{ver_CUDA[-1]}"
if os.name == "nt":
ver_TRITON += ".post25"
ver_TRITON += ".post26"
sage_package = os.environ.get("SAGE_PACKAGE", f"https://github.com/woct0rdho/SageAttention/releases/download/v{ver_SAGE}-windows.post4/sageattention-{ver_SAGE}+{ver_CUDA}torch2.9.0andhigher.post4-cp39-abi3-win_amd64.whl")
flash_package = os.environ.get("FLASH_PACKAGE", f"https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.13/flash_attn-{ver_FLASH}+{ver_CUDA}torch{v_TORCH}-{ver_PY}-{ver_PY}-win_amd64.whl")
flash_package = os.environ.get("FLASH_PACKAGE", f"https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.6/flash_attn-{ver_FLASH}+{ver_CUDA}torch{v_TORCH}-{ver_PY}-{ver_PY}-win_amd64.whl")
triton_package = os.environ.get("TRITION_PACKAGE", f"triton-windows=={ver_TRITON}")
nunchaku_package = os.environ.get("NUNCHAKU_PACKAGE", f"https://github.com/nunchaku-ai/nunchaku/releases/download/v{ver_NUNCHAKU}/nunchaku-{ver_NUNCHAKU}+{v_CUDA}torch{v_TORCH}-{ver_PY}-{ver_PY}-win_amd64.whl")
else:
sage_package = os.environ.get("SAGE_PACKAGE", f"sageattention=={ver_SAGE}")
flash_package = os.environ.get("FLASH_PACKAGE", f"https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.16/flash_attn-{ver_FLASH}+{ver_CUDA}torch{v_TORCH}-{ver_PY}-{ver_PY}-linux_x86_64.whl")
flash_package = os.environ.get("FLASH_PACKAGE", f"https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.4/flash_attn-{ver_FLASH}+{ver_CUDA}torch{v_TORCH}-{ver_PY}-{ver_PY}-linux_x86_64.whl")
triton_package = os.environ.get("TRITION_PACKAGE", f"triton=={ver_TRITON}")
nunchaku_package = os.environ.get("NUNCHAKU_PACKAGE", f"https://github.com/nunchaku-ai/nunchaku/releases/download/v{ver_NUNCHAKU}/nunchaku-{ver_NUNCHAKU}+{v_CUDA}torch{v_TORCH}-{ver_PY}-{ver_PY}-linux_x86_64.whl")
@ -434,7 +432,7 @@ assert cuda or xpu or mps
if args.onnxruntime_gpu and not is_installed("onnxruntime-gpu"):
# https://onnxruntime.ai/docs/install/#nightly-for-cuda-13x
_deps = "flatbuffers numpy packaging protobuf sympy coloredlogs"
_deps = "coloredlogs flatbuffers numpy packaging protobuf sympy"
onnxruntime_package = os.environ.get("ONNX_PACKAGE", "onnxruntime-gpu --pre --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-13-nightly/pypi/simple/")
run_pip(f"install {_deps}", "onnxruntime dependencies")
run_pip(f"install {onnxruntime_package}", "onnxruntime-gpu")

16
requirements.txt
View File

@ -1,16 +1,16 @@
GitPython==3.1.46
Pillow==12.1.0
accelerate==1.12.0
Pillow==12.2.0
accelerate==1.13.0
audioop-lts==0.2.2;python_version>="3.13"
av==17.0.0
comfy-kitchen==0.2.8
diffusers==0.36.0
diffusers==0.37.1
diskcache==5.6.3
einops==0.8.2
facexlib==0.3.0
fastapi==0.127.1
httpx==0.24.1
huggingface-hub==0.34.6
huggingface-hub==0.36.2
inflection==0.5.1
joblib==1.5.3
kornia==0.6.12
@ -21,19 +21,19 @@ omegaconf==2.2.3
opencv-python==4.10.0.84
peft==0.17.1
piexif==1.1.3
pillow-heif==1.2.0
pillow-heif==1.3.0
pillow-jxl-plugin==1.3.7
protobuf==4.25.8
protobuf==4.25.9
psutil==6.1.1
pydantic-core==2.27.2
pydantic==2.10.6
pyyaml==6.0.3
rich==14.3.2
rich==14.3.3
safetensors==0.7.0
scikit-image==0.25.2
setuptools==69.5.1
spandrel-extra-arches==0.2.0
spandrel==0.4.1
spandrel==0.4.2
tomesd==0.1.3
torchdiffeq==0.2.5
torchsde==0.2.6