diff options
Diffstat (limited to 'config-model/src/main/resources/schema/common.rnc')
-rw-r--r-- | config-model/src/main/resources/schema/common.rnc | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/config-model/src/main/resources/schema/common.rnc b/config-model/src/main/resources/schema/common.rnc index 2f3b10742f3..919253977ca 100644 --- a/config-model/src/main/resources/schema/common.rnc +++ b/config-model/src/main/resources/schema/common.rnc @@ -80,7 +80,7 @@ ComponentDefinition = TypedComponentDefinition = attribute id { xsd:Name } & - (HuggingFaceEmbedder | HuggingFaceTokenizer | BertBaseEmbedder | ColBertEmbedder) & + (HuggingFaceEmbedder | HuggingFaceTokenizer | BertBaseEmbedder | ColBertEmbedder | SpladeEmbedder ) & GenericConfig* & Component* @@ -97,6 +97,18 @@ HuggingFaceEmbedder = OnnxModelExecutionParams & EmbedderPoolingStrategy +SpladeEmbedder = + attribute type { "splade-embedder" } & + element transformer-model { ModelReference } & + element tokenizer-model { ModelReference }? & + element max-tokens { xsd:positiveInteger }? & + 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 term-score-threshold { xsd:double }? & + OnnxModelExecutionParams + HuggingFaceTokenizer = attribute type { "hugging-face-tokenizer" } & element model { attribute language { xsd:string }? & ModelReference }+ |