Fix config issue to support Triton

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · dae5eb0a4a
1 changed files with 3 additions and 1 deletions
--- 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):