vumichien commited on
Commit
8fb55c9
1 Parent(s): 40ca095
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ sub_test_list = [i.strip() for i in sub_test_list]
42
  embeddings = torch.load('embeddings.pt')
43
  # Configure
44
  options = onnxruntime.SessionOptions()
45
- options.intra_op_num_threads = 2
46
- options.inter_op_num_threads = 2
47
  # Load model
48
  PATH = 'model_onnx.onnx'
49
  ort_session = onnxruntime.InferenceSession(PATH)
 
42
  embeddings = torch.load('embeddings.pt')
43
  # Configure
44
  options = onnxruntime.SessionOptions()
45
+ options.intra_op_num_threads = 1
46
+ options.inter_op_num_threads = 1
47
  # Load model
48
  PATH = 'model_onnx.onnx'
49
  ort_session = onnxruntime.InferenceSession(PATH)