kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

Set TORCH_HOME

Browse Source
This commit is contained in:
loretoparisi 2021-11-15 23:37:26 +01:00
parent c8e3604f71
commit 7114f988b4
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@ -26,6 +26,7 @@ os.environ["OPENBLAS_NUM_THREADS"] = NUM_THREADS
os.environ["MKL_NUM_THREADS"] = NUM_THREADS os.environ["MKL_NUM_THREADS"] = NUM_THREADS
os.environ["VECLIB_MAXIMUM_THREADS"] = NUM_THREADS os.environ["VECLIB_MAXIMUM_THREADS"] = NUM_THREADS
os.environ["NUMEXPR_NUM_THREADS"] = NUM_THREADS os.environ["NUMEXPR_NUM_THREADS"] = NUM_THREADS
if os.environ["cache_dir"]:
os.environ["TORCH_HOME"] = 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") BUILD_DIR = os.environ.get("LAMA_CLEANER_BUILD_DIR", "./lama_cleaner/app/build")