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

HF dependencies

Browse Source
This commit is contained in:
Henk
2023-08-23 20:34:40 +02:00
parent f66173f2a0
commit 91155ed2f3
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
environments/huggingface.yml
View File

@@ -34,7 +34,8 @@ dependencies:
- lupa==1.10
- transformers==4.32.*
- huggingface_hub==0.16.4
- git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06
- optimum==1.12.0
- safetensors==0.3.2
- accelerate==0.21.0
- git+https://github.com/VE-FORBRYDERNE/mkultra
- flask-session

3
environments/rocm.yml
View File

@@ -32,7 +32,8 @@ dependencies:
- lupa==1.10
- transformers==4.32.*
- huggingface_hub==0.16.4
- git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06
- optimum==1.12.0
- safetensors==0.3.2
- accelerate==0.21.0
- git+https://github.com/VE-FORBRYDERNE/mkultra
- ansi2html

3
requirements.txt
View File

@@ -1,6 +1,7 @@
transformers==4.32.*
huggingface_hub==0.16.4
git+https://github.com/huggingface/optimum@2c1eaf6c0242badb86f3d626811ee3d7d220ec06
optimum==1.12.0
safetensors==0.3.2
Flask==2.2.3
Flask-SocketIO==5.3.2
python-socketio==5.7.2