Browse Source
Support onnx in tritonserve
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
4 additions and
0 deletions
-
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 = [ |
|
|
|