echarlaix's picture
echarlaix HF Staff
Add stsb-bert-tiny exported to OpenVINO (no PyTorch weights) (#1)
3b4e11c
Raw History Blame Contribute Delete
1.23 kB
{
"schema_version": 1,
"export_format": "openvino",
"components": {
"model": {
"file": "openvino_model.xml",
"metadata": {
"schema_version": 1,
"architecture": "_ExportableModel",
"packages": {
"transformers": "5.19.0.dev0",
"torch": "2.13.0",
"openvino": "2026.4.1"
},
"dtype": "float32",
"device": "cpu",
"kwargs": {
"input_ids": {
"rank": 2,
"shape": [
2,
16
],
"dtype": "int64"
},
"token_type_ids": {
"rank": 2,
"shape": [
2,
16
],
"dtype": "int64"
},
"attention_mask": {
"rank": 2,
"shape": [
2,
16
],
"dtype": "int64"
}
},
"input_names": [
"input_ids",
"token_type_ids",
"attention_mask"
],
"constant_inputs": {},
"output_names": [
"token_embeddings",
"sentence_embedding"
],
"num_user_outputs": 2
}
}
}
}