summaryrefslogtreecommitdiffstats
path: root/config-model/src/test/cfg
diff options
context:
space:
mode:
authorJo Kristian Bergum <bergum@yahooinc.com>2023-09-21 19:31:36 +0200
committerJo Kristian Bergum <bergum@yahooinc.com>2023-09-21 19:31:36 +0200
commit3062f88ca79cfa802d0a0c4e36de55d1f9e8245d (patch)
tree93642a0e36c848a9a9e04ab51666ebf6296a954d /config-model/src/test/cfg
parentd04a31c05a399df95717accfd656d4c68d78e611 (diff)
Add config options + license
Diffstat (limited to 'config-model/src/test/cfg')
-rw-r--r--config-model/src/test/cfg/application/embed/services.xml19
1 files changed, 19 insertions, 0 deletions
diff --git a/config-model/src/test/cfg/application/embed/services.xml b/config-model/src/test/cfg/application/embed/services.xml
index 70eef7ea54a..efb33d36761 100644
--- a/config-model/src/test/cfg/application/embed/services.xml
+++ b/config-model/src/test/cfg/application/embed/services.xml
@@ -43,6 +43,25 @@
<onnx-gpu-device>1</onnx-gpu-device>
</component>
+ <component id="colbert" type="colbert-embedder">
+ <transformer-model model-id="e5-base-v2" url="https://my/url/model.onnx"/>
+ <tokenizer-model model-id="e5-base-v2-vocab" path="app/tokenizer.json"/>
+ <max-tokens>1024</max-tokens>
+ <max-query-tokens>32</max-query-tokens>
+ <max-document-tokens>512</max-document-tokens>
+ <transformer-start-sequence-token>101</transformer-start-sequence-token>
+ <transformer-end-sequence-token>102</transformer-end-sequence-token>
+ <transformer-mask-token>103</transformer-mask-token>
+ <transformer-input-ids>my_input_ids</transformer-input-ids>
+ <transformer-attention-mask>my_attention_mask</transformer-attention-mask>
+ <transformer-token-type-ids>my_token_type_ids</transformer-token-type-ids>
+ <transformer-output>my_output</transformer-output>
+ <onnx-execution-mode>parallel</onnx-execution-mode>
+ <onnx-intraop-threads>10</onnx-intraop-threads>
+ <onnx-interop-threads>8</onnx-interop-threads>
+ <onnx-gpu-device>1</onnx-gpu-device>
+ </component>
+
<nodes>
<node hostalias="node1" />
</nodes>