update entrypoint and force min accelerate

2023-05-18 06:25:34 -04:00
parent 13650732f8
commit fa8bd14be4
2 changed files with 6 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ PyYAML==6.0
 black
 bitsandbytes==0.37.2
 datasets
-accelerate
+accelerate>=0.19.0
 sentencepiece
 wandb
 einops