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 - 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

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

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