Skip to content
This repository was archived by the owner on Dec 4, 2025. It is now read-only.

Commit 1cb69a9

Browse files
committed
Add constructor in OnnxBertBiEncoder
Signed-off-by: Jorge Bescos Gascon <[email protected]>
1 parent 7ef08e1 commit 1cb69a9

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

langchain4j-embeddings/src/main/java/dev/langchain4j/model/embedding/onnx/OnnxBertBiEncoder.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,14 @@ public OnnxBertBiEncoder(InputStream model, InputStream tokenizer, PoolingMode p
4242
}
4343
}
4444

45+
public OnnxBertBiEncoder(OrtEnvironment environment, OrtSession session, HuggingFaceTokenizer tokenizer, PoolingMode poolingMode) {
46+
this.environment = environment;
47+
this.session = session;
48+
this.expectedInputs = session.getInputNames();
49+
this.tokenizer = tokenizer;
50+
this.poolingMode = poolingMode;
51+
}
52+
4553
static class EmbeddingAndTokenCount {
4654

4755
float[] embedding;

0 commit comments

Comments
 (0)