Require accelerate

Transformers 4.20 now requires accelerate to be installed for some of the features we use in KoboldAI. This is now a required dependency for updated users.
2022-06-16 20:38:58 +02:00 · 2022-06-16 20:38:58 +02:00 · 49a3cf132e
parent 3504581015
commit 49a3cf132e
3 changed files with 4 additions and 1 deletions
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@ -20,4 +20,5 @@ dependencies:
    - flask-cloudflared
    - flask-ngrok
    - lupa==1.10
-    - transformers>=4.20
+    - transformers>=4.20
+    - accelerate
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@ -21,3 +21,4 @@ dependencies:
    - flask-ngrok
    - lupa==1.10
    - transformers>=4.20
+    - accelerate
--- a/requirements.txt
+++ b/requirements.txt
@ -11,3 +11,4 @@ markdown
 bleach==4.1.0
 sentencepiece
 protobuf
+accelerate