wip-4b3

2026-08-07 06:55:46 -07:00
parent 211b8f8c81
commit 09df5235e0
2 changed files with 16 additions and 1 deletions
@@ -75,3 +75,6 @@ libhdf5-compare = ["hdf5"]
 mpi-io = ["clawhdf5-io/mpi-io", "mpi"]
 # Real MiniLM embeddings for longmemeval_bench, so the vector stage is not inert.
 embeddings = ["candle-core", "candle-nn", "candle-transformers", "tokenizers"]
+# CUDA-accelerated embedding. MiniLM on a CPU takes hours over the full
+# longmemeval_s haystack; on a GPU it is minutes.
+embeddings-cuda = ["embeddings", "candle-core/cuda", "candle-nn/cuda", "candle-transformers/cuda"]