diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index 9fd2ce5f..485ac338 100644
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@@ -22,7 +22,6 @@ dependencies:
   - loguru
   - termcolor
   - Pillow
-  - diffusers
   - psutil
   - pip:
     - flask-cloudflared
@@ -40,4 +39,5 @@ dependencies:
     - ijson
     - bitsandbytes
     - ftfy
-    - pydub
\ No newline at end of file
+    - pydub
+    - diffusers
diff --git a/environments/rocm.yml b/environments/rocm.yml
index ad39a3a6..a0e23177 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -19,12 +19,10 @@ dependencies:
   - loguru
   - termcolor
   - Pillow
-  - diffusers
   - psutil
   - pip:
     - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
-    - torch
-    - torchvision
+    - torch==1.11.*
     - flask-cloudflared
     - flask-ngrok
     - lupa==1.10
@@ -38,3 +36,4 @@ dependencies:
     - ijson
     - ftfy
     - pydub
+    - diffusers