diff --git a/modules/launch_utils.py b/modules/launch_utils.py index 5500382a..59f73788 100644 --- a/modules/launch_utils.py +++ b/modules/launch_utils.py @@ -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"