diff --git a/modules/launch_utils.py b/modules/launch_utils.py index 20c7dc127..5ca8c01b8 100644 --- a/modules/launch_utils.py +++ b/modules/launch_utils.py @@ -317,7 +317,7 @@ def requirements_met(requirements_file): def prepare_environment(): torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu121") - torch_command = os.environ.get('TORCH_COMMAND', f"pip install torch==2.1.2 torchvision==0.16.2 --extra-index-url {torch_index_url}") + torch_command = os.environ.get('TORCH_COMMAND', f"pip install torch==2.9.1 torchvision==0.24.1 --extra-index-url {torch_index_url}") if args.use_ipex: if platform.system() == "Windows": # The "Nuullll/intel-extension-for-pytorch" wheels were built from IPEX source for Intel Arc GPU: https://github.com/intel/intel-extension-for-pytorch/tree/xpu-main diff --git a/webui-user.bat b/webui-user.bat index e5a257bef..ddbbdd691 100644 --- a/webui-user.bat +++ b/webui-user.bat @@ -3,6 +3,8 @@ set PYTHON= set GIT= set VENV_DIR= -set COMMANDLINE_ARGS= +set COMMANDLINE_ARGS=--skip-python-version-check --skip-torch-cuda-test --skip-install +set STABLE_DIFFUSION_REPO=https://github.com/CompVis/stable-diffusion.git +set STABLE_DIFFUSION_COMMIT_HASH=main call webui.bat