diff options
author | Jon Bratseth <bratseth@vespa.ai> | 2024-02-02 12:28:53 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@vespa.ai> | 2024-02-02 12:28:53 +0100 |
commit | 1a25431ab58c752c7fc26dd8223bf1ba1079b24a (patch) | |
tree | 954d7e2f3e43bb0636a6af7a93195a84e41e609b /config-model | |
parent | 2191193c6e107eb68611ddb106e5f572bea32903 (diff) |
Support embedding into rank 3 tensors
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java b/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java index 9e0a3a0ba5c..92c930e16e0 100644 --- a/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java +++ b/config-model/src/main/java/com/yahoo/vespa/model/container/component/SpladeEmbedder.java @@ -61,5 +61,6 @@ public class SpladeEmbedder extends TypedComponent implements SpladeEmbedderConf onnxModelOptions.intraOpThreads().ifPresent(b::transformerIntraOpThreads); onnxModelOptions.gpuDevice().ifPresent(value -> b.transformerGpuDevice(value.deviceNumber())); } + } |