From e2b82b9e17993e8c72f86e34b8b1831a661920bc Mon Sep 17 00:00:00 2001
From: Henk <henk@henk.tech>
Date: Sun, 8 Jan 2023 15:31:12 +0100
Subject: [PATCH] ROCm Dependency Fixes

---
 environments/huggingface.yml | 4 ++--
 environments/rocm.yml        | 5 ++---
 2 files changed, 4 insertions(+), 5 deletions(-)

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