diff options
-rw-r--r-- | model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def b/model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def index 14f5e95a6b8..ef42d81e1fe 100644 --- a/model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def +++ b/model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def @@ -18,8 +18,8 @@ transformerAttentionMask string default=attention_mask transformerTokenTypeIds string default=token_type_ids # special token ids -transformerStartSequenceToken int default= -transformerEndSequenceToken int default= +transformerStartSequenceToken int default=101 +transformerEndSequenceToken int default=102 # Output name transformerOutput string default=output_0 |