aboutsummaryrefslogtreecommitdiffstats
path: root/launch.py
diff options
context:
space:
mode:
Diffstat (limited to 'launch.py')
-rw-r--r--launch.py15
1 files changed, 1 insertions, 14 deletions
diff --git a/launch.py b/launch.py
index 668548f1..bcbb792c 100644
--- a/launch.py
+++ b/launch.py
@@ -7,7 +7,6 @@ import shlex
import platform
import argparse
import json
-import detection
dir_repos = "repositories"
dir_extensions = "extensions"
@@ -16,12 +15,6 @@ git = os.environ.get('GIT', "git")
index_url = os.environ.get('INDEX_URL', "")
stored_commit_hash = None
-# Get the GPU vendor and the operating system
-gpu = detection.check_gpu()
-if os.name == "posix":
- os_name = platform.uname().system
-else:
- os_name = os.name
def commit_hash():
global stored_commit_hash
@@ -180,11 +173,7 @@ def run_extensions_installers(settings_file):
def prepare_environment():
- if gpu == "AMD" and os_name !="nt":
- torch_command = os.environ.get('TORCH_COMMAND', "pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2")
- else:
- torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113")
-
+ torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113")
requirements_file = os.environ.get('REQS_FILE', "requirements_versions.txt")
commandline_args = os.environ.get('COMMANDLINE_ARGS', "")
@@ -306,8 +295,6 @@ def tests(test_dir):
def start():
- print(f"Operating System: {os_name}")
- print(f"GPU: {gpu}")
print(f"Launching {'API server' if '--nowebui' in sys.argv else 'Web UI'} with arguments: {' '.join(sys.argv[1:])}")
import webui
if '--nowebui' in sys.argv: