From 976368836d226b3fb1369eeebfcc1ddeb070575a Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 22 Jun 2022 11:20:55 +0800 Subject: [PATCH] Update todo Signed-off-by: Jael Gu --- auto_transformers.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/auto_transformers.py b/auto_transformers.py index c3f1120..a85feeb 100644 --- a/auto_transformers.py +++ b/auto_transformers.py @@ -128,9 +128,7 @@ class AutoTransformers(NNOperator): "last_hidden_state": {0: "batch_size"}, "pooler_outputs": {0: "batch_size"} }) - elif format == 'tensorrt': - # os.system('pip install "git+https://github.com/grimoire/torch2trt_dynamic.git"') - pass + # todo: elif format == 'tensorrt': else: log.error(f'Unsupported format "{format}".') @@ -371,6 +369,7 @@ class AutoTransformers(NNOperator): ] assert set(to_remove).issubset(set(full_list)) model_list = list(set(full_list) - set(to_remove)) - else: # todo: format in {'tensorrt'} + # todo: elif format == 'tensorrt': + else: log.error(f'Invalid format "{format}". Currently supported formats: "pytorch", "torchscript".') return model_list