rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2024-12-12 08:36:28 +01:00
Code Issues Projects Releases Wiki Activity

Requirements Updates

Browse Source
This commit is contained in:
Henk 2022-09-25 19:33:18 +02:00
parent 6009586ac8
commit 9f15077337
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
environments/rocm.yml
View File

@ -18,8 +18,8 @@ dependencies:
- apispec-webframeworks - apispec-webframeworks
- loguru - loguru
- pip: - pip:
- --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
- torch==1.10.* - torch
- torchvision - torchvision
- flask-cloudflared - flask-cloudflared
- flask-ngrok - flask-ngrok

2
requirements.txt
View File

@ -2,7 +2,7 @@ transformers>=4.20.1
Flask Flask
Flask-SocketIO Flask-SocketIO
requests requests
torch==1.11 torch<=1.12.*
flask-cloudflared flask-cloudflared
flask-ngrok flask-ngrok
eventlet eventlet

2
requirements_mtj.txt
View File

@ -1,4 +1,4 @@
torch >= 1.9, <= 1.11 torch >= 1.9, <= 1.12.*
numpy numpy
tqdm tqdm
requests requests