ROCm Dependency Fixes

This commit is contained in:
Henk
2023-01-08 15:31:12 +01:00
parent 2d85046493
commit e2b82b9e17
2 changed files with 4 additions and 5 deletions

View File

@@ -22,7 +22,6 @@ dependencies:
- loguru - loguru
- termcolor - termcolor
- Pillow - Pillow
- diffusers
- psutil - psutil
- pip: - pip:
- flask-cloudflared - flask-cloudflared
@@ -40,4 +39,5 @@ dependencies:
- ijson - ijson
- bitsandbytes - bitsandbytes
- ftfy - ftfy
- pydub - pydub
- diffusers

View File

@@ -19,12 +19,10 @@ dependencies:
- loguru - loguru
- termcolor - termcolor
- Pillow - Pillow
- diffusers
- psutil - psutil
- pip: - pip:
- --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
- torch - torch==1.11.*
- torchvision
- flask-cloudflared - flask-cloudflared
- flask-ngrok - flask-ngrok
- lupa==1.10 - lupa==1.10
@@ -38,3 +36,4 @@ dependencies:
- ijson - ijson
- ftfy - ftfy
- pydub - pydub
- diffusers