From f48644a285e8e08a55dcb64f6132053df7cca601 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Sun, 29 Jan 2023 16:13:55 +0800 Subject: [PATCH] Remove failed models Signed-off-by: Jael Gu --- auto_transformers.py | 2 -- test_onnx.py | 5 +++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/auto_transformers.py b/auto_transformers.py index 995b14e..056a98c 100644 --- a/auto_transformers.py +++ b/auto_transformers.py @@ -233,8 +233,6 @@ class AutoTransformers(NNOperator): 'bert-large-uncased', 'bert-large-uncased-whole-word-masking', 'distilbert-base-uncased', - 'facebook/dpr-ctx_encoder-multiset-base', - 'facebook/dpr-ctx_encoder-single-nq-base', 'facebook/bart-large', 'gpt2-xl', 'microsoft/deberta-xlarge', diff --git a/test_onnx.py b/test_onnx.py index c209feb..58134d4 100644 --- a/test_onnx.py +++ b/test_onnx.py @@ -17,7 +17,8 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' warnings.filterwarnings("ignore") t_logging.set_verbosity_error() -# full_models = AutoTransformers.supported_model_names() +# op = ops.sentence_embedding.transformers().get_op() +# full_models = op.supported_model_names() # checked_models = AutoTransformers.supported_model_names(format='onnx') # models = [x for x in full_models if x not in checked_models] models = ['distilbert-base-cased', 'sentence-transformers/paraphrase-albert-small-v2'] @@ -80,7 +81,7 @@ for name in models: status[3] = 'success' except Exception as e: logger.error(f'FAIL TO CHECK ONNX: {e}') - continue + pass try: sess = onnxruntime.InferenceSession(onnx_path, providers=onnxruntime.get_available_providers())