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 = [