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())