mirror of
https://github.com/ollama/ollama.git
synced 2026-04-24 17:55:43 +02:00
use float32
This commit is contained in:
@@ -2,18 +2,18 @@ package format
|
||||
|
||||
import "math"
|
||||
|
||||
func Normalize(vec []float64) []float64 {
|
||||
func Normalize(vec []float32) []float32 {
|
||||
var sum float64
|
||||
for _, v := range vec {
|
||||
sum += v * v
|
||||
sum += float64(v * v)
|
||||
}
|
||||
|
||||
sum = math.Sqrt(sum)
|
||||
|
||||
var norm float64
|
||||
var norm float32
|
||||
|
||||
if sum > 0 {
|
||||
norm = 1.0 / sum
|
||||
norm = float32(1.0 / sum)
|
||||
} else {
|
||||
norm = 0.0
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user