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

Pytorch 2.0

Browse Source
This commit is contained in:
Henk
2023-05-05 13:14:58 +02:00
parent a87d5d6f23
commit 33745669dd
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
environments/huggingface.yml
View File

@@ -1,6 +1,7 @@
name: koboldai name: koboldai
channels: channels:
- pytorch - pytorch
- nvidia
- conda-forge - conda-forge
- defaults - defaults
dependencies: dependencies:
@@ -9,9 +10,9 @@ dependencies:
- flask-socketio=5.3.2 - flask-socketio=5.3.2
- flask-session=0.4.0 - flask-session=0.4.0
- python-socketio=5.7.2 - python-socketio=5.7.2
- pytorch=1.11.* - pytorch=2.0.*
- python=3.8.* - python=3.8.*
- cudatoolkit=11.1 - pytorch-cuda=11.8
- eventlet=0.33.3 - eventlet=0.33.3
- dnspython=2.2.1 - dnspython=2.2.1
- markdown - markdown
@@ -45,3 +46,4 @@ dependencies:
- ftfy - ftfy
- pydub - pydub
- diffusers - diffusers
- peft

3
requirements.txt
View File

@@ -4,7 +4,7 @@ Flask==2.2.3
Flask-SocketIO==5.3.2 Flask-SocketIO==5.3.2
python-socketio==5.7.2 python-socketio==5.7.2
requests requests
torch >= 1.9, < 1.13 torch == 2.0.*
flask-cloudflared==0.0.10 flask-cloudflared==0.0.10
flask-ngrok flask-ngrok
flask-cors flask-cors
@@ -37,3 +37,4 @@ pytest-html==3.2.0
pytest-metadata==2.0.4 pytest-metadata==2.0.4
requests-mock==1.10.0 requests-mock==1.10.0
safetensors safetensors
peft