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):