From 2b75ca9ecaf8781c5e48afcabd77336dc077eb5b Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 8 Jan 2023 20:42:19 -0500 Subject: [PATCH] update launcher --- automatic.sh | 3 ++- launch.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/automatic.sh b/automatic.sh index 88a0881a0..9ae467a08 100755 --- a/automatic.sh +++ b/automatic.sh @@ -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 diff --git a/launch.py b/launch.py index 49b91b1f2..6256832f4 100644 --- a/launch.py +++ b/launch.py @@ -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")