Merge branch 'KoboldAI:main' into united

2022-06-16 20:42:24 +02:00 · 2022-06-16 20:42:24 +02:00 · b0a01962ab
parent 50d2172aaf 49a3cf132e
commit b0a01962ab
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