Spaces:
Runtime error
Runtime error
Commit
·
b975c77
1
Parent(s):
3167459
set to max thread
Browse files
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)
|