diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 669d5d6d..2ac51431 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -33,7 +33,7 @@ dependencies: - lupa==1.10 - transformers==4.28.0 - huggingface_hub==0.12.1 - - safetensors + - safetensors==0.3.1 - accelerate==0.18.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session diff --git a/environments/rocm.yml b/environments/rocm.yml index dc2dd40e..1329612b 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -32,7 +32,7 @@ dependencies: - lupa==1.10 - transformers==4.28.0 - huggingface_hub==0.12.1 - - safetensors + - safetensors==0.3.1 - accelerate==0.18.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - ansi2html diff --git a/requirements.txt b/requirements.txt index 6407303c..cb567b08 100644 --- a/requirements.txt +++ b/requirements.txt @@ -36,4 +36,4 @@ pytest==7.2.2 pytest-html==3.2.0 pytest-metadata==2.0.4 requests-mock==1.10.0 -safetensors \ No newline at end of file +safetensors==0.3.1