From 7fc5c46c1d38a5bbf0c902556743148aab983ed8 Mon Sep 17 00:00:00 2001 From: Henk Date: Fri, 6 Jan 2023 16:40:56 +0100 Subject: [PATCH] Add Safetensors Having the dependency adds basic support for safetensor models. --- environments/huggingface.yml | 1 + environments/rocm.yml | 1 + requirements.txt | 1 + requirements_mtj.txt | 1 + 4 files changed, 4 insertions(+) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index dbd2b8c4..3f25a7fd 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -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 diff --git a/environments/rocm.yml b/environments/rocm.yml index 75143858..f47dd6c1 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -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 diff --git a/requirements.txt b/requirements.txt index 4eb50b03..44f50bdd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -18,4 +18,5 @@ marshmallow>=3.13 apispec-webframeworks loguru termcolor +safetensors git+https://github.com/VE-FORBRYDERNE/mkultra \ No newline at end of file diff --git a/requirements_mtj.txt b/requirements_mtj.txt index c2c61432..5d84dd43 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -20,4 +20,5 @@ bleach==4.1.0 flask-session marshmallow>=3.13 apispec-webframeworks +safetensors loguru