logo
Browse Source

Support onnx in tritonserve

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
c2dbfd95e5
  1. 4
      auto_transformers.py

4
auto_transformers.py

@ -229,6 +229,10 @@ class AutoTransformers(NNOperator):
else: else:
return ['pytorch'] return ['pytorch']
@property
def supported_formats(self):
return ['onnx']
@staticmethod @staticmethod
def supported_model_names(format: str = None): def supported_model_names(format: str = None):
full_list = [ full_list = [

Loading…
Cancel
Save