更新 requirements.txt

yuhuixu1993 · Oct 9, 2023 · 489c8a2 · 489c8a2
1 parent f3d03e3
commit 489c8a2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,3 +4,6 @@ rouge-score==0.1.2
 scikit-learn==1.2.2
 sentencepiece==0.1.99
 wandb==0.15.2
+transformers==4.31.0
+peft==0.4.0
+accelerate==0.21.0