summaryrefslogtreecommitdiffstats
path: root/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def
diff options
context:
space:
mode:
Diffstat (limited to 'model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def')
-rw-r--r--model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def1
1 files changed, 1 insertions, 0 deletions
diff --git a/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def b/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def
index 1dccea0ddf6..97515818f14 100644
--- a/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def
+++ b/model-integration/src/main/resources/configdefinitions/hugging-face-embedder.def
@@ -12,6 +12,7 @@ transformerMaxTokens int default=512
# Input names
transformerInputIds string default=input_ids
transformerAttentionMask string default=attention_mask
+transformerTokenTypeIds string default=token_type_ids
# Output name
transformerOutput string default=last_hidden_state