diff --git a/lama_cleaner/server.py b/lama_cleaner/server.py index 769692f..5b44592 100644 --- a/lama_cleaner/server.py +++ b/lama_cleaner/server.py @@ -40,6 +40,9 @@ from lama_cleaner.helper import ( 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["OPENBLAS_NUM_THREADS"] = NUM_THREADS os.environ["MKL_NUM_THREADS"] = NUM_THREADS