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

Pin accelerate version

Browse Source
This commit is contained in:
Henk
2023-04-15 17:45:00 +02:00
parent 3eda7269f7
commit 67334bd698
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
environments/rocm.yml
View File

@@ -32,7 +32,7 @@ dependencies:
- transformers==4.28.0 - transformers==4.28.0
- huggingface_hub==0.12.1 - huggingface_hub==0.12.1
- safetensors - safetensors
- accelerate - accelerate==0.18.0
- git+https://github.com/VE-FORBRYDERNE/mkultra - git+https://github.com/VE-FORBRYDERNE/mkultra
- ansi2html - ansi2html
- flask_compress - flask_compress

2
requirements.txt
View File

@@ -15,7 +15,7 @@ markdown
bleach==4.1.0 bleach==4.1.0
sentencepiece sentencepiece
protobuf protobuf
accelerate accelerate==0.18.0
flask-session==0.4.0 flask-session==0.4.0
marshmallow>=3.13 marshmallow>=3.13
apispec-webframeworks apispec-webframeworks