We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0499ea5 commit 4c50015Copy full SHA for 4c50015
fastembed/common/onnx_model.py
@@ -85,7 +85,7 @@ def _load_onnx_model(
85
86
so = ort.SessionOptions()
87
so.graph_optimization_level = ort.GraphOptimizationLevel.ORT_ENABLE_ALL
88
-
+ so.add_session_config_entry("memory.enable_memory_arena_shrinkage", "1")
89
if threads is not None:
90
so.intra_op_num_threads = threads
91
so.inter_op_num_threads = threads
0 commit comments