ROCm Dependency Fixes

2025-06-05 21:59:24 +02:00 · 2023-01-08 15:31:12 +01:00
parent 2d85046493
commit e2b82b9e17
2 changed files with 4 additions and 5 deletions
--- 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
+    - pydub
+    - diffusers
--- 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