1 files changed, 3 insertions, 2 deletions
diff --git a/requirements.txt b/requirements.txt
index 29f3cbd..d583e6e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
+accelerate
 datasets
-evaluate
 torch
-transformers
-\ No newline at end of file
+transformers
+wandb
+\ No newline at end of file