ashishraics commited on
Commit
b975c77
·
1 Parent(s): 3167459

set to max thread

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -93,8 +93,8 @@ st.markdown(hide_streamlit_style, unsafe_allow_html=True)
93
  #onnx runtime inference threading changes -- session options must be passed in session run
94
  # os.environ["OMP_NUM_THREADS"] = "1" #use this before changing session options of onnx runtime
95
  session_options_ort = ort.SessionOptions()
96
- session_options_ort.intra_op_num_threads=1
97
- session_options_ort.inter_op_num_threads=1
98
  # session_options_ort.execution_mode = session_options_ort.ExecutionMode.ORT_SEQUENTIAL
99
 
100
  @st.cache(allow_output_mutation=True, suppress_st_warning=True, max_entries=None, ttl=None)
 
93
  #onnx runtime inference threading changes -- session options must be passed in session run
94
  # os.environ["OMP_NUM_THREADS"] = "1" #use this before changing session options of onnx runtime
95
  session_options_ort = ort.SessionOptions()
96
+ session_options_ort.intra_op_num_threads=total_threads #1
97
+ session_options_ort.inter_op_num_threads=total_threads #1
98
  # session_options_ort.execution_mode = session_options_ort.ExecutionMode.ORT_SEQUENTIAL
99
 
100
  @st.cache(allow_output_mutation=True, suppress_st_warning=True, max_entries=None, ttl=None)