Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
1 year ago · 03c85991e3
1 changed files with 2 additions and 1 deletions
--- a/timm_image.py
+++ b/timm_image.py
@ -189,7 +189,8 @@ class TimmImage(NNOperator):

    @staticmethod
    def supported_model_names(format: str = None):
-        assert timm.__version__ == '0.6.12', 'The model lists are tested with timm==0.6.12.'
+        if timm.__version__ != '0.6.12':
+            log.warning('Please note that the model list is tested with timm==0.6.12, please check your timm version.')
        full_list = list(set(timm.list_models(pretrained=True)) - set([
            'coat_mini',
            'coat_tiny',