diff --git a/examples/advanced/vertical_federated_learning/cifar10-splitnn/requirements.txt b/examples/advanced/vertical_federated_learning/cifar10-splitnn/requirements.txt index 914c7b5958..83c759685b 100644 --- a/examples/advanced/vertical_federated_learning/cifar10-splitnn/requirements.txt +++ b/examples/advanced/vertical_federated_learning/cifar10-splitnn/requirements.txt @@ -1,6 +1,6 @@ -nvflare~=2.5.0rc +nvflare~=2.5.0 torch torchvision tensorboard -openmined.psi==1.1.1 +openmined.psi==2.0.5 pandas diff --git a/research/fed-bpt/README.md b/research/fed-bpt/README.md index e32a3ea144..67f63449d6 100644 --- a/research/fed-bpt/README.md +++ b/research/fed-bpt/README.md @@ -14,7 +14,7 @@ The models code is copied from the [transformers](https://github.com/huggingface We recommend creating a [conda environment](https://www.anaconda.com) following [BBT](https://github.com/txsun1997/Black-Box-Tuning#prepare-your-environment) with the addition of installing NVFlare for running federated learning and some other updates: ```commandline -conda create --name fedbpt python=3.8 +conda create --name fedbpt python=3.12 conda activate fedbpt pip install -r requirements.txt ``` diff --git a/research/fed-bpt/requirements.txt b/research/fed-bpt/requirements.txt index c840dde822..f2757c231f 100644 --- a/research/fed-bpt/requirements.txt +++ b/research/fed-bpt/requirements.txt @@ -1,8 +1,9 @@ transformers==4.38.2 fastNLP==0.6.0 datasets -cma +cma==3.4.0 scikit-learn tensorboard cvxopt -nvflare~=2.4.1rc \ No newline at end of file +nvflare~=2.5.1rc +