Disabled TensorrtExecutionProvider due to a bug, it will now run with CUDAExecutionProvider on NVIDIA until we resolve error #9 #10

This commit is contained in:
Xavi Vinaixa
2023-06-06 12:39:20 +02:00
parent 4e833ec3d1
commit f2e4dd93f4

View File

@@ -56,6 +56,8 @@ class Refacer:
self.mode = RefacerMode.CUDA
self.use_num_cpus = 1
self.sess_options.intra_op_num_threads = 1
if 'TensorrtExecutionProvider' in self.providers:
self.providers.remove('TensorrtExecutionProvider')
print(f"CUDA mode with providers {self.providers}")
"""
elif 'TensorrtExecutionProvider' in self.providers: