From dae5eb0a4ae38ff8c918a973d15ea293f944345a Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Tue, 27 Dec 2022 16:03:21 +0800 Subject: [PATCH] Fix config issue to support Triton Signed-off-by: Jael Gu --- timm_image.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/timm_image.py b/timm_image.py index f6888cf..3d68bb3 100644 --- a/timm_image.py +++ b/timm_image.py @@ -120,7 +120,9 @@ class TimmImage(NNOperator): @property def config(self): - return resolve_data_config({}, model=self._model) + m = create_model(self.model_name, pretrained=False) + config = resolve_data_config({}, model=m) + return config @arg(1, to_image_color('RGB')) def convert_img(self, img: towhee._types.Image):