diff --git a/environments/huggingface.yml b/environments/huggingface.yml index a5affd40..9fd2ce5f 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -30,6 +30,7 @@ dependencies: - lupa==1.10 - transformers==4.24.0 - huggingface_hub>=0.10.1 + - safetensors - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session diff --git a/environments/rocm.yml b/environments/rocm.yml index 2318b123..ad39a3a6 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -30,6 +30,7 @@ dependencies: - lupa==1.10 - transformers==4.24.0 - huggingface_hub>=0.10.1 + - safetensors - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra - ansi2html diff --git a/requirements.txt b/requirements.txt index c1429449..53e450c9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,4 +27,5 @@ flask_compress ijson bitsandbytes ftfy -pydub \ No newline at end of file +pydub +safetensors \ No newline at end of file diff --git a/requirements_mtj.txt b/requirements_mtj.txt index 2a103144..39c8451e 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -20,6 +20,7 @@ bleach==4.1.0 flask-session marshmallow>=3.13 apispec-webframeworks +safetensors loguru Pillow diffusers