Skip to content

Commit 64ac32c

Browse files
2 parents 2314843 + dfeeb87 commit 64ac32c

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

BabbleApp/babble_processor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,8 +119,8 @@ def __init__(
119119
self.sess = ort.InferenceSession(
120120
f"{self.default_model}/onnx/model.onnx",
121121
self.opts,
122-
providers=[provider],
123-
provider_options=[{"device_id": self.gpu_index}],
122+
providers=provider,
123+
provider_options=[{"device_id": self.gpu_index}, {"device_id": self.gpu_index}, {}], # We need a dict entry for each EP in our providers list
124124
)
125125
self.input_name = self.sess.get_inputs()[0].name
126126
self.output_name = self.sess.get_outputs()[0].name

0 commit comments

Comments
 (0)