瀏覽代碼

check normalization

Roy Han 10 月之前
父節點
當前提交
b9c74df37b
共有 1 個文件被更改,包括 9 次插入0 次删除
  1. 9 0
      server/routes.go

+ 9 - 0
server/routes.go

@@ -480,6 +480,15 @@ func (s *Server) EmbeddingsHandler(c *gin.Context) {
 		return
 	}
 
+	// assert that embedding is normalized
+	sum := 0.0
+	for _, v := range embedding {
+		sum += v * v
+	}
+	if math.Abs(sum-1) > 1e-6 {
+		slog.Info("embedding is not normalized", "sum", sum)
+	}
+
 	resp := api.EmbeddingResponse{
 		Embedding: embedding,
 	}