From 28108ecb73402ced03a3f3d0c6bd4a4477a07748 Mon Sep 17 00:00:00 2001 From: ddPn08 Date: Mon, 17 Jul 2023 22:20:09 +0900 Subject: [PATCH] fix launch.py --- launch.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/launch.py b/launch.py index 1a98654..f627c6b 100644 --- a/launch.py +++ b/launch.py @@ -96,7 +96,7 @@ def prepare_environment(): torch_command = os.environ.get( "TORCH_COMMAND", - "pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118", + "pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118", ) sys.argv, skip_install = extract_arg(sys.argv, "--skip-install") @@ -106,10 +106,10 @@ def prepare_environment(): sys.argv, reinstall_torch = extract_arg(sys.argv, "--reinstall-torch") ngrok = "--ngrok" in sys.argv - if reinstall_torch or not is_installed("torch") or not is_installed("torchvision"): + if reinstall_torch or not is_installed("torch") or not is_installed("torchaudio"): run( f'"{python}" -m {torch_command}', - "Installing torch and torchvision", + "Installing torch and torchaudio", "Couldn't install torch", )