rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-01-10 07:20:04 +01:00
Code Issues Projects Releases Wiki Activity

Add Safetensors

Browse Source 
Having the dependency adds basic support for safetensor models.
This commit is contained in:
Henk 2023-01-06 16:40:56 +01:00
parent 1dbc987048
commit 7fc5c46c1d
4 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
environments/huggingface.yml
View File

@ -28,5 +28,6 @@ dependencies:
- lupa==1.10
- transformers==4.24.0
- huggingface_hub>=0.10.1
- safetensors
- accelerate
- git+https://github.com/VE-FORBRYDERNE/mkultra

1
environments/rocm.yml
View File

@ -28,5 +28,6 @@ dependencies:
- lupa==1.10
- transformers==4.24.0
- huggingface_hub>=0.10.1
- safetensors
- accelerate
- git+https://github.com/VE-FORBRYDERNE/mkultra

1
requirements.txt
View File

@ -18,4 +18,5 @@ marshmallow>=3.13
apispec-webframeworks
loguru
termcolor
safetensors
git+https://github.com/VE-FORBRYDERNE/mkultra

1
requirements_mtj.txt
View File

@ -20,4 +20,5 @@ bleach==4.1.0
flask-session
marshmallow>=3.13
apispec-webframeworks
safetensors
loguru