devve1 commited on
Commit
dead9ef
1 Parent(s): 846c662

Update optimum_encoder.py

Browse files
Files changed (1) hide show
  1. optimum_encoder.py +1 -1
optimum_encoder.py CHANGED
@@ -72,7 +72,7 @@ class OptimumEncoder(BaseEncoder):
72
  model_id=self.name,
73
  file_name='model_fp16.onnx',
74
  subfolder='onnx',
75
- provider=['TensorrtExecutionProvider'],
76
  provider_options=provider_options,
77
  session_options=session_options,
78
  **self.model_kwargs
 
72
  model_id=self.name,
73
  file_name='model_fp16.onnx',
74
  subfolder='onnx',
75
+ provider='TensorrtExecutionProvider',
76
  provider_options=provider_options,
77
  session_options=session_options,
78
  **self.model_kwargs