update launcher

This commit is contained in:
Vladimir Mandic
2023-01-08 20:42:19 -05:00
parent bbb08beb48
commit 2b75ca9eca
2 changed files with 4 additions and 3 deletions
+2 -1
View File
@@ -1,3 +1,4 @@
#/bin/env bash
export PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.9,max_split_size_mb:512
python launch.py --api --api-log --xformers --disable-console-progressbars
python launch.py --api --xformers --disable-console-progressbars
+2 -2
View File
@@ -211,8 +211,8 @@ def prepare_environment():
commit = commit_hash()
print(f"Python {sys.version}")
print(f"Commit hash: {commit}")
# print(f"Python {sys.version}")
# print(f"Commit hash: {commit}")
if not is_installed("torch") or not is_installed("torchvision"):
run(f'"{python}" -m {torch_command}', "Installing torch and torchvision", "Couldn't install torch")