Skip to content
This repository was archived by the owner on Feb 20, 2026. It is now read-only.

Commit e36d735

Browse files
committed
fix pca transform loading
1 parent b93034a commit e36d735

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

ols_embed/server.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -140,8 +140,10 @@ def embed():
140140

141141
# Apply PCA transformation if needed
142142
if use_pca and pca_transformer is not None:
143-
# PCA transformer expects the data in the same format it was trained on
144-
embeddings_array = pca_transformer.transform(embeddings_array.astype(np.float64))
143+
# PCA transformer is a dict with 'mean_' and 'components_'
144+
# Transform: (X - mean) @ components
145+
embeddings_array = embeddings_array.astype(np.float64)
146+
embeddings_array = (embeddings_array - pca_transformer['mean_']) @ pca_transformer['components_']
145147
embeddings_array = embeddings_array.astype(np.float32)
146148

147149
num_vectors, dimension = embeddings_array.shape

0 commit comments

Comments
 (0)