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)
|