FedBPT: Fix fedbpt cma version (#3029) (#3030)

* fix cma version; upgrade nvflare version * upgrade python to 3.12
NVIDIA · Oct 10, 2024 · 737d2ea · 737d2ea
1 parent 2001852
commit 737d2ea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
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
+