diff --git a/timm_image.py b/timm_image.py
index 8668525..bc2ff0e 100644
--- 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'