From 9eb3da12c2fb8b1aa6da43c17741528c8dbcd87c Mon Sep 17 00:00:00 2001 From: Lajcisvk Date: Sat, 21 Oct 2023 01:09:11 +0200 Subject: [PATCH] Apply thread limit to intra_op_num_threads (#529) --- rembg/session_factory.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rembg/session_factory.py b/rembg/session_factory.py index ae7f009..bd59939 100644 --- a/rembg/session_factory.py +++ b/rembg/session_factory.py @@ -39,5 +39,6 @@ def new_session( if "OMP_NUM_THREADS" in os.environ: sess_opts.inter_op_num_threads = int(os.environ["OMP_NUM_THREADS"]) + sess_opts.intra_op_num_threads = int(os.environ["OMP_NUM_THREADS"]) return session_class(model_name, sess_opts, providers, *args, **kwargs)