Browse Source
Remove failed models
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
2 changed files with
3 additions and
4 deletions
-
auto_transformers.py
-
test_onnx.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', |
|
|
|
|
|
@ -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()) |
|
|
|