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

Dependencies required for newer models

Browse Source
This commit is contained in:
henk717
2022-02-21 21:17:12 +01:00
parent 4ace11f5b8
commit cb6ccacd64
4 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
environments/huggingface.yml
View File

@ -9,13 +9,15 @@ dependencies:
- pytorch
- python=3.8.*
- cudatoolkit=11.1
- transformers
- eventlet
- markdown
- bleach
- pip
- git
- sentencepiece
- protobuf
- pip:
- flask-cloudflared
- flask-ngrok
- lupa==1.10
- git+https://github.com/huggingface/transformers

4
environments/rocm.yml
View File

@ -3,7 +3,6 @@ channels:
- conda-forge
- defaults
dependencies:
- transformers
- colorama
- flask-socketio
- python=3.8.*
@ -12,6 +11,8 @@ dependencies:
- bleach
- pip
- git
- sentencepiece
- protobuf
- pip:
- --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html
- torch
@ -19,3 +20,4 @@ dependencies:
- flask-cloudflared
- flask-ngrok
- lupa==1.10
- git+https://github.com/huggingface/transformers

1
fairseq-dense-2.7B-Janeway Submodule

Submodule fairseq-dense-2.7B-Janeway added at 99ec323435

4
requirements.txt
View File

@ -8,4 +8,6 @@ flask-ngrok
eventlet
lupa==1.10
markdown
bleach
bleach
sentencepiece
protobuf