Merge branch 'main' into enhance_json_serialization_for_numpy

YuanTingHsieh · Oct 10, 2024 · 1a5f9ec · 1a5f9ec
2 parents 178b518 + 791a70b
commit 1a5f9ec
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
+nvflare~=2.5.1rc
+