diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index da11ce10..d65f9446 100644
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@@ -20,4 +20,5 @@ dependencies:
     - flask-cloudflared
     - flask-ngrok
     - lupa==1.10
-    - transformers>=4.20
\ No newline at end of file
+    - transformers>=4.20
+    - accelerate
\ No newline at end of file
diff --git a/environments/rocm.yml b/environments/rocm.yml
index 1fd66d26..d3e10515 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -21,3 +21,4 @@ dependencies:
     - flask-ngrok
     - lupa==1.10
     - transformers>=4.20
+    - accelerate
diff --git a/requirements.txt b/requirements.txt
index 67c6ce74..e6308790 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,3 +11,4 @@ markdown
 bleach==4.1.0
 sentencepiece
 protobuf
+accelerate
\ No newline at end of file