Merge branch 'KoboldAI:main' into united

2025-06-05 21:59:24 +02:00 · 2022-06-16 20:42:24 +02:00
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