improve fine-tune performance

2024-02-04 19:33:32 +08:00
parent e3e075e432
commit 0da92ec7bf
2 changed files with 4 additions and 3 deletions
--- a/finetune/requirements.txt
+++ b/finetune/requirements.txt
@@ -1,3 +1,3 @@
-torch==1.13.1
+torch==2.1.2
 pytorch_lightning==1.9.5
-deepspeed==0.11.2
+deepspeed==0.12.6