New intent classification engine - MiniLM L6v2 and MiniLM L12v2 ONNX
This commit is contained in:
parent
4815c7f9bb
commit
8e830334e8
25 changed files with 61932 additions and 150 deletions
|
|
@ -0,0 +1,39 @@
|
|||
{
|
||||
"one_external_file": true,
|
||||
"opset": null,
|
||||
"optimization": {
|
||||
"disable_attention": null,
|
||||
"disable_attention_fusion": false,
|
||||
"disable_bias_gelu": null,
|
||||
"disable_bias_gelu_fusion": false,
|
||||
"disable_bias_skip_layer_norm": null,
|
||||
"disable_bias_skip_layer_norm_fusion": false,
|
||||
"disable_embed_layer_norm": true,
|
||||
"disable_embed_layer_norm_fusion": true,
|
||||
"disable_gelu": null,
|
||||
"disable_gelu_fusion": false,
|
||||
"disable_group_norm_fusion": true,
|
||||
"disable_layer_norm": null,
|
||||
"disable_layer_norm_fusion": false,
|
||||
"disable_packed_kv": true,
|
||||
"disable_rotary_embeddings": false,
|
||||
"disable_shape_inference": false,
|
||||
"disable_skip_layer_norm": null,
|
||||
"disable_skip_layer_norm_fusion": false,
|
||||
"enable_gelu_approximation": true,
|
||||
"enable_gemm_fast_gelu_fusion": false,
|
||||
"enable_transformers_specific_optimizations": true,
|
||||
"fp16": true,
|
||||
"no_attention_mask": false,
|
||||
"optimization_level": 2,
|
||||
"optimize_for_gpu": true,
|
||||
"optimize_with_onnxruntime_only": null,
|
||||
"use_mask_index": false,
|
||||
"use_multi_head_attention": false,
|
||||
"use_raw_attention_mask": false
|
||||
},
|
||||
"optimum_version": "1.15.0",
|
||||
"quantization": {},
|
||||
"transformers_version": "4.36.2",
|
||||
"use_external_data_format": false
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue