diff --git a/dependencies/requirements.gpu.txt b/dependencies/requirements.gpu.txt index 1eeaea2..20dcc3e 100644 --- a/dependencies/requirements.gpu.txt +++ b/dependencies/requirements.gpu.txt @@ -14,3 +14,10 @@ sentence-transformers==2.5.1 faiss_cpu==1.8.0 linformer-pytorch==0.19.3 PyGithub==2.3.0 +deap==1.4.1 +update_checker==0.18.0 +scikit-mdr==0.4.5 +skrebate==0.62 +xgboost==2.0.3 +stopit==1.1.2 +tpot==0.12.2