Spaces:
Running
on
T4
Running
on
T4
Update optimum_encoder.py
Browse files- 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=
|
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
|