diff options
Diffstat (limited to 'model-integration/src/main/java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/embedding/SpladeEmbedder.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/embedding/SpladeEmbedder.java b/model-integration/src/main/java/ai/vespa/embedding/SpladeEmbedder.java index 4b90fa0a9bf..28f8c4e252f 100644 --- a/model-integration/src/main/java/ai/vespa/embedding/SpladeEmbedder.java +++ b/model-integration/src/main/java/ai/vespa/embedding/SpladeEmbedder.java @@ -158,7 +158,7 @@ public class SpladeEmbedder extends AbstractComponent implements Embedder { double maxValue = 0.0d; directAddress.setIndex(2, v); long increment = directAddress.getStride(1); - long directIndex = directAddress.getIndex(); + long directIndex = directAddress.getDirectIndex(); for (int s = 0; s < sequenceLength; s++) { double value = modelOutput.get(directIndex + s * increment); if (value > maxValue) { |