diff options
| author | Grail Finder <wohilas@gmail.com> | 2026-03-05 14:49:59 +0300 |
|---|---|---|
| committer | Grail Finder <wohilas@gmail.com> | 2026-03-05 14:49:59 +0300 |
| commit | c2757653a3429ab3f9e76081328a3877bc11ed4d (patch) | |
| tree | 51684e9dba28fe97dde0dcd64299476213619494 | |
| parent | 4bd6883966824cff81b86e8bf79e278165d7d24a (diff) | |
Fix: buildable
| -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 { |