logo
Browse Source

Fix config issue to support Triton

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
dae5eb0a4a
  1. 4
      timm_image.py

4
timm_image.py

@ -120,7 +120,9 @@ class TimmImage(NNOperator):
@property @property
def config(self): 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')) @arg(1, to_image_color('RGB'))
def convert_img(self, img: towhee._types.Image): def convert_img(self, img: towhee._types.Image):

Loading…
Cancel
Save