From c2dbfd95e5f355e6397dc34e9dbf3f88014b5a99 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Thu, 12 Jan 2023 14:25:28 +0800
Subject: [PATCH] Support onnx in tritonserve

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 auto_transformers.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/auto_transformers.py b/auto_transformers.py
index 496f989..1ce8b7d 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -229,6 +229,10 @@ class AutoTransformers(NNOperator):
         else:
             return ['pytorch']
 
+    @property
+    def supported_formats(self):
+        return ['onnx']
+
     @staticmethod
     def supported_model_names(format: str = None):
         full_list = [