fix bug

microsoft · Sep 19, 2023 · 7c5c374 · 7c5c374
1 parent d5936a2
commit 7c5c374
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/python/onnxruntime_inference_collection.py b/onnxruntime/python/onnxruntime_inference_collection.py
@@ -522,7 +522,7 @@ def _register_ep_custom_ops(self, session_options, providers, provider_options):
         for i in range(len(providers)):
             if providers[i] == "TensorrtExecutionProvider":
                 C.register_tensorrt_plugins_as_custom_ops(session_options, provider_options[i])
-            elif (isinstance(providers[i], tuple) and providers[0] == "TensorrtExecutionProvider"): 
+            elif (isinstance(providers[i], tuple) and providers[i][0] == "TensorrtExecutionProvider"): 
                 C.register_tensorrt_plugins_as_custom_ops(session_options, providers[i][1])
 
 class IOBinding: