reqs for starchat?

1 year ago · 97d5190a7c
parent af90f4444d
commit 97d5190a7c
1 changed files with 3 additions and 0 deletions
--- a/programming_runs/requirements.txt
+++ b/programming_runs/requirements.txt
@ -3,3 +3,6 @@ openai==0.27.0
 datasets==2.7.0
 tenacity==8.1.0
 astunparse==1.6.3
+torch
+xformers
+transformers