rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Requirements Updates

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

4
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
@ -15,4 +15,4 @@ accelerate
flask-session flask-session
marshmallow>=3.13 marshmallow>=3.13
apispec-webframeworks apispec-webframeworks
loguru loguru

4
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
@ -19,4 +19,4 @@ bleach==4.1.0
flask-session flask-session
marshmallow>=3.13 marshmallow>=3.13
apispec-webframeworks apispec-webframeworks
loguru loguru