logo
Browse Source

Remove failed models

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
f48644a285
  1. 2
      auto_transformers.py
  2. 5
      test_onnx.py

2
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',

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

Loading…
Cancel
Save