fix some para

This commit is contained in:
yichuan520030910320
2025-07-23 01:46:34 -07:00
parent ae16d9d888
commit 851f0f04c3
2 changed files with 2 additions and 2 deletions

View File

@@ -101,7 +101,7 @@ def compute_embeddings_sentence_transformers(
if device == "mps":
batch_size = 128 # MPS optimal batch size from benchmark
if model_name == "Qwen/Qwen3-Embedding-0.6B":
batch_size = 64
batch_size = 32
elif device == "cuda":
batch_size = 256 # CUDA optimal batch size
# Keep original batch_size for CPU