From 8da18d800a9880e108415c0e41e20271538946b6 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 19 Dec 2022 15:42:28 +0800 Subject: [PATCH] Fix issue for config Signed-off-by: Jael Gu --- timm_image.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/timm_image.py b/timm_image.py index 8ddea6f..d54676a 100644 --- a/timm_image.py +++ b/timm_image.py @@ -117,7 +117,7 @@ class TimmImage(NNOperator): @property def config(self): - return resolve_data_config({}, model=self.model) + return resolve_data_config({}, model=self._model) @arg(1, to_image_color('RGB')) def convert_img(self, img: towhee._types.Image): @@ -195,14 +195,7 @@ class TimmImage(NNOperator): 'crossvit_base_240', 'crossvit_small_240', 'crossvit_tiny_240', - 'tresnet_l', - 'tresnet_l_448', - 'tresnet_m', - 'tresnet_m_448', - 'tresnet_m_miil_in21k', - 'tresnet_v2_l', - 'tresnet_xl', - 'tresnet_xl_448'] + ] assert set(to_remove).issubset(set(full_list)) model_list = list(set(full_list) - set(to_remove)) elif format == 'onnx':