From f944b96338725a0a75bbe52922f98f9342abcdd4 Mon Sep 17 00:00:00 2001 From: Bjørn Christian Seime Date: Mon, 5 Jun 2023 15:31:33 +0200 Subject: Move config definition to `configdefinitions` --- .../embedding.bert-base-embedder.def | 32 ---------------------- 1 file changed, 32 deletions(-) delete mode 100644 model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def (limited to 'model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def') 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 deleted file mode 100644 index 2d8e840377b..00000000000 --- a/model-integration/src/main/resources/configdefinitions/embedding.bert-base-embedder.def +++ /dev/null @@ -1,32 +0,0 @@ - -namespace=embedding - -# Wordpiece tokenizer -tokenizerVocab model - -transformerModel model - -# Max length of token sequence model can handle -transformerMaxTokens int default=384 - -# Pooling strategy -poolingStrategy enum { cls, mean } default=mean - -# Input names -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 - -# Settings for ONNX model evaluation -onnxExecutionMode enum { parallel, sequential } default=sequential -onnxInterOpThreads int default=1 -onnxIntraOpThreads int default=-4 # n=number of threads -> n<0: CPUs/(-n), n==0: CPUs, n>0: n -# GPU device id, -1 for CPU -onnxGpuDevice int default=0 -- cgit v1.2.3