From 7114f988b4cefaa66f55dbaf12e7e700d32954d1 Mon Sep 17 00:00:00 2001 From: loretoparisi Date: Mon, 15 Nov 2021 23:37:26 +0100 Subject: [PATCH] Set TORCH_HOME --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index b52ac21..780e999 100644 --- a/main.py +++ b/main.py @@ -26,7 +26,8 @@ os.environ["OPENBLAS_NUM_THREADS"] = NUM_THREADS os.environ["MKL_NUM_THREADS"] = NUM_THREADS os.environ["VECLIB_MAXIMUM_THREADS"] = NUM_THREADS os.environ["NUMEXPR_NUM_THREADS"] = NUM_THREADS -os.environ["TORCH_HOME"] = os.environ["cache_dir"] +if os.environ["cache_dir"]: + os.environ["TORCH_HOME"] = os.environ["cache_dir"] BUILD_DIR = os.environ.get("LAMA_CLEANER_BUILD_DIR", "./lama_cleaner/app/build")