diff options
| -rw-r--r-- | rag/embedder.go | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/rag/embedder.go b/rag/embedder.go index 988d91e..6903a5d 100644 --- a/rag/embedder.go +++ b/rag/embedder.go @@ -215,7 +215,6 @@ func (e *ONNXEmbedder) Embed(text string) ([]float32, error) { // 5. Run inference err = e.session.Run( []onnxruntime_go.Value{inputIDsTensor, maskTensor}, - []string{"sentence_embedding"}, []onnxruntime_go.Value{outputTensor}, ) if err != nil { @@ -270,7 +269,6 @@ func (e *ONNXEmbedder) EmbedSlice(texts []string) ([][]float32, error) { defer outputTensor.Destroy() err := e.session.Run( []onnxruntime_go.Value{inputTensor, maskTensor}, - []string{"sentence_embedding"}, []onnxruntime_go.Value{outputTensor}, ) if err != nil { |