Use float32 for embeddings

hupe1980 · Dec 15, 2023 · 1ebc3e4 · 1ebc3e4
1 parent ebfd86c
commit 1ebc3e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/feature_extraction.go b/feature_extraction.go
@@ -15,10 +15,10 @@ type FeatureExtractionRequest struct {
 }
 
 // Response structure for the feature extraction endpoint
-type FeatureExtractionResponse [][][][]float64
+type FeatureExtractionResponse [][][][]float32
 
 // Response structure for the feature extraction endpoint
-type FeatureExtractionWithAutomaticReductionResponse [][]float64
+type FeatureExtractionWithAutomaticReductionResponse [][]float32
 
 // FeatureExtraction performs feature extraction using the specified model.
 // It sends a POST request to the Hugging Face inference endpoint with the provided input data.