Tags: bclavie/RAGatouille
Tags
Feat/indexing faissless (#173) * fix: fix searcher always being reloaded * feat: implement torch kmeans * chore: lower cutoff * chore: move warning * chore: higher kmeans batch size * chore: argument support * chore: restore all default behaviour when `use_faiss` is True after having been false * chore: lint * chore: print exception if one occurs when using pytorch indexing * chore: make _original_train_kmeans robust to subsequent calls * nit: comment feat: rework kmeans to be closer to FAISS chore: store kmeans functions as class attributes fix: method assignment chore: more memory efficient lint chore: lower bsize, resultd unaffected feat: better batching, slower max doc count chore: batch size safe for 8gb GPUs chore: more elaborate warning chore: use external lib to support minibatching, revert to homebrew later * poetry lock * lint * chore: better batch size * 0.0.8 dependency prep
PreviousNext