diff options
author | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-06-05 17:17:01 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-06-05 17:51:00 +0200 |
commit | b2e9037a14c2865d8c6377f9de3e07ad06627d9d (patch) | |
tree | 3204c48c4d3a69119801bdc83d13e5e36a196aae /configdefinitions | |
parent | 684eb28e3b38ce015f5af060a5ba7b14f8f999c9 (diff) |
Make pooling strategy configurable for Huggingface embedder
Diffstat (limited to 'configdefinitions')
-rw-r--r-- | configdefinitions/src/vespa/hugging-face-embedder.def | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/configdefinitions/src/vespa/hugging-face-embedder.def b/configdefinitions/src/vespa/hugging-face-embedder.def index 36957004e02..7ea4227b3cd 100644 --- a/configdefinitions/src/vespa/hugging-face-embedder.def +++ b/configdefinitions/src/vespa/hugging-face-embedder.def @@ -21,6 +21,8 @@ transformerOutput string default=last_hidden_state # Normalize tensors from tokenizer normalize bool default=false +poolingStrategy enum { cls, mean } default=mean + # Settings for ONNX model evaluation transformerExecutionMode enum { parallel, sequential } default=sequential transformerInterOpThreads int default=1 |