From dae5eb0a4ae38ff8c918a973d15ea293f944345a Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Tue, 27 Dec 2022 16:03:21 +0800
Subject: [PATCH] Fix config issue to support Triton

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 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):