grail/gf-lt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: buildable

Browse Source
This commit is contained in:
Grail Finder
2026-03-05 14:49:59 +03:00
parent 4bd6883966
commit c2757653a3
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/embedder.go
View File

@@ -215,7 +215,6 @@ func (e *ONNXEmbedder) Embed(text string) ([]float32, error) {
// 5. Run inference // 5. Run inference
err = e.session.Run( err = e.session.Run(
[]onnxruntime_go.Value{inputIDsTensor, maskTensor}, []onnxruntime_go.Value{inputIDsTensor, maskTensor},
[]string{"sentence_embedding"},
[]onnxruntime_go.Value{outputTensor}, []onnxruntime_go.Value{outputTensor},
) )
if err != nil { if err != nil {
@@ -270,7 +269,6 @@ func (e *ONNXEmbedder) EmbedSlice(texts []string) ([][]float32, error) {
defer outputTensor.Destroy() defer outputTensor.Destroy()
err := e.session.Run( err := e.session.Run(
[]onnxruntime_go.Value{inputTensor, maskTensor}, []onnxruntime_go.Value{inputTensor, maskTensor},
[]string{"sentence_embedding"},
[]onnxruntime_go.Value{outputTensor}, []onnxruntime_go.Value{outputTensor},
) )
if err != nil { if err != nil {