Set TORCH_HOME

This commit is contained in:
loretoparisi 2021-11-15 23:37:26 +01:00
parent c8e3604f71
commit 7114f988b4

View File

@ -26,6 +26,7 @@ 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
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")