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

fix https://github.com/Sanster/lama-cleaner/issues/56

Browse Source
This commit is contained in:
Qing 2022-07-26 09:22:27 +08:00
parent b6cae0ebf4
commit 9b95376a79
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lama_cleaner/server.py
View File

@ -40,6 +40,9 @@ from lama_cleaner.helper import (
NUM_THREADS = str(multiprocessing.cpu_count()) NUM_THREADS = str(multiprocessing.cpu_count())
# fix libomp problem on windows https://github.com/Sanster/lama-cleaner/issues/56
os.environ["KMP_DUPLICATE_LIB_OK"]="True"
os.environ["OMP_NUM_THREADS"] = NUM_THREADS os.environ["OMP_NUM_THREADS"] = NUM_THREADS
os.environ["OPENBLAS_NUM_THREADS"] = NUM_THREADS os.environ["OPENBLAS_NUM_THREADS"] = NUM_THREADS
os.environ["MKL_NUM_THREADS"] = NUM_THREADS os.environ["MKL_NUM_THREADS"] = NUM_THREADS