Apply thread limit to intra_op_num_threads (#529)

danielgatis · Oct 20, 2023 · 9eb3da1 · 9eb3da1
1 parent 0d0f6c6
commit 9eb3da1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)