From 6fe678adc5a1f30cdccf55ca462111dafe35c88b Mon Sep 17 00:00:00 2001 From: Haoming Date: Mon, 13 Apr 2026 14:06:23 +0800 Subject: [PATCH] deps --- modules/launch_utils.py | 16 +++++++--------- requirements.txt | 16 ++++++++-------- 2 files changed, 15 insertions(+), 17 deletions(-) diff --git a/modules/launch_utils.py b/modules/launch_utils.py index bc5527f2..32b77b8f 100644 --- a/modules/launch_utils.py +++ b/modules/launch_utils.py @@ -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") diff --git a/requirements.txt b/requirements.txt index 57104620..4aac8e14 100644 --- a/requirements.txt +++ b/requirements.txt @@ -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