diff options
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.def | 1 |
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 |