diff --git a/requirements.txt b/requirements.txt index 3da67e8..04f9623 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,4 +9,6 @@ tenacity==8.2.3 pyyaml==6.0.1 pydantic==2.5.2 transformers==4.36.2 -torch==2.1.2 \ No newline at end of file +torch==2.1.2 +scikit-learn>=1.3.0 +faiss-cpu>=1.7.4 \ No newline at end of file