fix: ark embedding norm

This commit is contained in:
xuzhaonan 2025-07-23 11:07:31 +08:00 committed by 徐兆楠
parent 5d1c55ab1a
commit dc984f8afb
1 changed files with 1 additions and 2 deletions

View File

@ -76,9 +76,8 @@ func (d embWrap) slicedNormL2(vectors [][]float64) ([][]float64, error) {
}
if curDims := len(vectors[0]); curDims < int(d.dims) {
return nil, fmt.Errorf("[slicedNormL2] got dims=%d less than %d", curDims, d.dims)
} else if curDims == int(d.dims) {
return vectors, nil
}
result := make([][]float64, len(vectors))
for i, vec := range vectors {
sliced := vec[:d.dims]