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

packaging

Browse Source
This commit is contained in:
Haoming 2025-09-24 09:48:00 +08:00
parent f150cacd6e
commit 84013acee2
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/launch_utils.py
View File

@ -285,6 +285,7 @@ def prepare_environment():
sage_package = os.environ.get("SAGE_PACKAGE", "sageattention==1.0.6")
clip_package = os.environ.get("CLIP_PACKAGE", "https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip")
packaging_package = os.environ.get("PACKAGING_PACKAGE", "packaging==24.2")
gradio_package = os.environ.get("GRADIO_PACKAGE", "gradio==4.40.0 gradio_imageslider==0.0.20 gradio_rangeslider==0.0.6")
requirements_file = os.environ.get("REQS_FILE", "requirements.txt")
@ -314,6 +315,9 @@ def prepare_environment():
raise RuntimeError("PyTorch is not able to access CUDA")
startup_timer.record("torch GPU test")
if not is_installed("packaging"):
run_pip(f"install {packaging_package}", "packaging")
ver_PY = f"cp{sys.version_info.major}{sys.version_info.minor}"
ver_FLASH = "2.8.2"
ver_TRITON = "3.4.0"