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")