Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 8c66702529
1 changed files with 1 additions and 1 deletions
--- a/timm_image.py
+++ b/timm_image.py
@ -256,7 +256,7 @@ class TimmImage(NNOperator):
                'volo_d5_448',
                'volo_d5_512'
            ]
            assert set(to_remove).issubset(set(full_list))
            # assert set(to_remove).issubset(set(full_list))
            model_list = list(set(full_list) - set(to_remove))
        # todo: elif format == 'torchscript':
        # todo: elif format == 'tensorrt'