diff options
| author | Jon Bratseth <bratseth@vespa.ai> | 2023-04-18 11:11:20 +0200 |
|---|---|---|
| committer | Jon Bratseth <bratseth@vespa.ai> | 2023-04-18 11:11:20 +0200 |
| commit | 5082e42350a117c46f045a31c3502f2c8ad97a0d (patch) | |
| tree | 58828b8922b16c19dc2746982a0a11b2e1dbf2ef /model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def | |
| parent | 95805d16d0b043f0149edf733876067f1a8ee8e7 (diff) | |
Revert "Merge pull request #26744 from vespa-engine/revert-26708-allow-start-end-sequence-tokens-as-args-bertbaseembedder"
This reverts commit d025a93015e66efc0027d81a64e70530d6cb240e, reversing
changes made to 4f2f29e1459b900d4b074f5cfc4c126837c54bfd.
Diffstat (limited to 'model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def')
| -rw-r--r-- | model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def | 4 |
1 files changed, 4 insertions, 0 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 14d953eeef9..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 @@ -17,6 +17,10 @@ transformerInputIds string default=input_ids transformerAttentionMask string default=attention_mask transformerTokenTypeIds string default=token_type_ids +# special token ids +transformerStartSequenceToken int default=101 +transformerEndSequenceToken int default=102 + # Output name transformerOutput string default=output_0 |