rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

ROCm torch version pin

Browse Source
This commit is contained in:
Henk
2023-01-08 17:10:59 +01:00
parent 7fc5c46c1d
commit 857476ef6b
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
environments/rocm.yml
View File

@@ -21,8 +21,7 @@ dependencies:
- 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