diff options
Diffstat (limited to 'config-model/src/main/resources/schema/common.rnc')
| -rw-r--r-- | config-model/src/main/resources/schema/common.rnc | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/config-model/src/main/resources/schema/common.rnc b/config-model/src/main/resources/schema/common.rnc index b2b71950a0c..061e54740f1 100644 --- a/config-model/src/main/resources/schema/common.rnc +++ b/config-model/src/main/resources/schema/common.rnc @@ -94,7 +94,8 @@ HuggingFaceEmbedder = element transformer-token-type-ids { xsd:string }? & element transformer-output { xsd:string }? & element normalize { xsd:boolean }? & - OnnxModelExecutionParams + OnnxModelExecutionParams & + EmbedderPoolingStrategy HuggingFaceTokenizer = attribute type { "hugging-face-tokenizer" } & @@ -108,17 +109,19 @@ BertBaseEmbedder = element transformer-model { ModelReference } & element tokenizer-vocab { ModelReference } & element max-tokens { xsd:nonNegativeInteger }? & - element pooling-strategy { "cls" | "mean" }? & element transformer-input-ids { xsd:string }? & element transformer-attention-mask { xsd:string }? & element transformer-token-type-ids { xsd:string }? & element transformer-output { xsd:string }? & element transformer-start-sequence-token { xsd:integer }? & element transformer-end-sequence-token { xsd:integer }? & - OnnxModelExecutionParams + OnnxModelExecutionParams & + EmbedderPoolingStrategy OnnxModelExecutionParams = element onnx-execution-mode { "parallel" | "sequential" }? & element onnx-interop-threads { xsd:integer }? & element onnx-intraop-threads { xsd:integer }? & - element onnx-gpu-device { xsd:integer }?
\ No newline at end of file + element onnx-gpu-device { xsd:integer }? + +EmbedderPoolingStrategy = element pooling-strategy { "cls" | "mean" }?
\ No newline at end of file |