Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · a04940fb30
1 changed files with 1 additions and 2 deletions
--- a/timm_image.py
+++ b/timm_image.py
@ -30,10 +30,9 @@ from timm.models.factory import create_model
 import warnings

 warnings.filterwarnings('ignore')
-
-
 log = logging.getLogger()

+
@register(output_schema=['vec'])
 class TimmImage(NNOperator):
    """