diff --git a/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/feature-extraction_fp16_config.json b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/feature-extraction_fp16_config.json new file mode 100644 index 000000000..e731b2a2a --- /dev/null +++ b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/feature-extraction_fp16_config.json @@ -0,0 +1,64 @@ +{ + "export": { + "opset_version": 17, + "batch_size": 1, + "export_params": true, + "do_constant_folding": true, + "verbose": false, + "dynamo": false, + "enable_hierarchy_tags": true, + "clean_onnx": false, + "hierarchy_tag_format": "full", + "input_tensors": [ + { + "name": "input_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 30527 + ] + }, + { + "name": "attention_mask", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + } + ], + "output_tensors": [ + { + "name": "last_hidden_state" + } + ] + }, + "optim": {}, + "quant": null, + "loader": { + "task": "feature-extraction", + "model_class": "AutoModel", + "model_type": "mpnet" + }, + "eval": { + "task": "feature-extraction", + "dataset": { + "path": "mteb/stsbenchmark-sts", + "split": "test", + "samples": 100, + "columns_mapping": { + "input_column_1": "sentence1", + "input_column_2": "sentence2", + "score_column": "score" + } + } + } +} diff --git a/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/feature-extraction_fp32_config.json b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/feature-extraction_fp32_config.json new file mode 100644 index 000000000..e731b2a2a --- /dev/null +++ b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/feature-extraction_fp32_config.json @@ -0,0 +1,64 @@ +{ + "export": { + "opset_version": 17, + "batch_size": 1, + "export_params": true, + "do_constant_folding": true, + "verbose": false, + "dynamo": false, + "enable_hierarchy_tags": true, + "clean_onnx": false, + "hierarchy_tag_format": "full", + "input_tensors": [ + { + "name": "input_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 30527 + ] + }, + { + "name": "attention_mask", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + } + ], + "output_tensors": [ + { + "name": "last_hidden_state" + } + ] + }, + "optim": {}, + "quant": null, + "loader": { + "task": "feature-extraction", + "model_class": "AutoModel", + "model_type": "mpnet" + }, + "eval": { + "task": "feature-extraction", + "dataset": { + "path": "mteb/stsbenchmark-sts", + "split": "test", + "samples": 100, + "columns_mapping": { + "input_column_1": "sentence1", + "input_column_2": "sentence2", + "score_column": "score" + } + } + } +} diff --git a/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/sentence-similarity_fp16_config.json b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/sentence-similarity_fp16_config.json new file mode 100644 index 000000000..25b22809c --- /dev/null +++ b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/sentence-similarity_fp16_config.json @@ -0,0 +1,64 @@ +{ + "export": { + "opset_version": 17, + "batch_size": 1, + "export_params": true, + "do_constant_folding": true, + "verbose": false, + "dynamo": false, + "enable_hierarchy_tags": true, + "clean_onnx": false, + "hierarchy_tag_format": "full", + "input_tensors": [ + { + "name": "input_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 30527 + ] + }, + { + "name": "attention_mask", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + } + ], + "output_tensors": [ + { + "name": "last_hidden_state" + } + ] + }, + "optim": {}, + "quant": null, + "loader": { + "task": "sentence-similarity", + "model_class": "AutoModel", + "model_type": "mpnet" + }, + "eval": { + "task": "sentence-similarity", + "dataset": { + "path": "mteb/stsbenchmark-sts", + "split": "test", + "samples": 100, + "columns_mapping": { + "input_column_1": "sentence1", + "input_column_2": "sentence2", + "score_column": "score" + } + } + } +} diff --git a/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/sentence-similarity_fp32_config.json b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/sentence-similarity_fp32_config.json new file mode 100644 index 000000000..25b22809c --- /dev/null +++ b/examples/recipes/dell-research-harvard_lt-wikidata-comp-en/cpu/cpu/sentence-similarity_fp32_config.json @@ -0,0 +1,64 @@ +{ + "export": { + "opset_version": 17, + "batch_size": 1, + "export_params": true, + "do_constant_folding": true, + "verbose": false, + "dynamo": false, + "enable_hierarchy_tags": true, + "clean_onnx": false, + "hierarchy_tag_format": "full", + "input_tensors": [ + { + "name": "input_ids", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 30527 + ] + }, + { + "name": "attention_mask", + "dtype": "int32", + "shape": [ + 1, + 512 + ], + "value_range": [ + 0, + 2 + ] + } + ], + "output_tensors": [ + { + "name": "last_hidden_state" + } + ] + }, + "optim": {}, + "quant": null, + "loader": { + "task": "sentence-similarity", + "model_class": "AutoModel", + "model_type": "mpnet" + }, + "eval": { + "task": "sentence-similarity", + "dataset": { + "path": "mteb/stsbenchmark-sts", + "split": "test", + "samples": 100, + "columns_mapping": { + "input_column_1": "sentence1", + "input_column_2": "sentence2", + "score_column": "score" + } + } + } +}