jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Roy Han 2024-07-03 11:20:55 -07:00
parent a5f23d766e
commit a413014aaf
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/routes.go
View File

@ -482,14 +482,9 @@ func normalize(vec []float32) []float32 {
sum += float64(v * v) sum += float64(v * v)
} }
sum = math.Sqrt(sum) norm := float32(0.0)
var norm float32
if sum > 0 { if sum > 0 {
norm = float32(1.0 / sum) norm = float32(1.0 / math.Sqrt(sum))
} else {
norm = 0.0
} }
for i := range vec { for i := range vec {