Merge branch 'main' into united

2025-06-05 21:59:24 +02:00 · 2023-01-06 16:55:18 +01:00
parent b4c2fa0dc1 7fc5c46c1d
commit 8fad03431d
4 changed files with 5 additions and 1 deletions
--- 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
--- 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
--- a/requirements.txt
+++ b/requirements.txt
@@ -28,3 +28,4 @@ ijson
 bitsandbytes
 ftfy
 pydub
+safetensors
--- 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