Signed-off-by: weijingchen <talkingwallace@sohu.com>

update requirement.txt Signed-off-by: cwj <talkingwallace@sohu.com>
FederatedAI · Dec 21, 2023 · ee8af64 · ee8af64
1 parent 9fee434
commit ee8af64
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/fate/ml/nn/trainer/trainer_base.py b/python/fate/ml/nn/trainer/trainer_base.py
@@ -125,7 +125,7 @@ class _TrainingArguments(_hf_TrainingArguments):
     log_level: str = field(default="info")
     deepspeed: Optional[str] = field(default=None)
     save_safetensors: bool = field(default=False)
-    use_cpu: bool = field(default=False)
+    use_cpu: bool = field(default=True)
 
 
     def __post_init__(self):

diff --git a/python/requirements-fate.txt b/python/requirements-fate.txt
@@ -9,8 +9,8 @@ click
 ruamel-yaml==0.16
 numpy
 pandas==2.0.3
-transformers==4.30.2
-accelerate==0.20.2
+transformers==4.36.0
+accelerate==0.25.0
 beautifultable
 requests<2.26.0
 scikit-learn