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

Dependencies required for newer models

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
environments/huggingface.yml
View File

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

4
environments/rocm.yml
View File

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

1
fairseq-dense-2.7B-Janeway Submodule

@ -0,0 +1 @@
Subproject commit 99ec3234354e8ac12f5a0a594469a815407ab3ab

4
requirements.txt
View File

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