diff options
| author | connell gough <connell@portrait-analytics.com> | 2023-04-11 09:47:37 -0700 |
|---|---|---|
| committer | connell gough <connell@portrait-analytics.com> | 2023-04-13 10:19:36 -0700 |
| commit | 5f157a6eedc43e4ddb19c99aeea721a93d740f89 (patch) | |
| tree | 8959997c7365aab624d8301131a9f4bcd48c8371 /model-integration | |
| parent | d398c0760456b8bffb35cf30c0edd708dcde69d4 (diff) | |
Add default special tokens
Diffstat (limited to 'model-integration')
| -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 |