From b2e9037a14c2865d8c6377f9de3e07ad06627d9d Mon Sep 17 00:00:00 2001 From: Bjørn Christian Seime Date: Mon, 5 Jun 2023 17:17:01 +0200 Subject: Make pooling strategy configurable for Huggingface embedder --- configdefinitions/src/vespa/hugging-face-embedder.def | 2 ++ 1 file changed, 2 insertions(+) (limited to 'configdefinitions') 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 -- cgit v1.2.3