diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 1e54011b..e053654a 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -35,7 +35,7 @@ dependencies: - transformers==4.32.* - huggingface_hub==0.16.4 - optimum==1.12.0 - - safetensors==0.3.2 + - safetensors==0.3.3 - accelerate==0.21.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session diff --git a/environments/rocm.yml b/environments/rocm.yml index 22aff85c..bd468a6a 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -33,7 +33,7 @@ dependencies: - transformers==4.32.* - huggingface_hub==0.16.4 - optimum==1.12.0 - - safetensors==0.3.2 + - safetensors==0.3.3 - accelerate==0.21.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - ansi2html diff --git a/requirements.txt b/requirements.txt index 4c9c6681..b7abbed1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ transformers==4.32.* huggingface_hub==0.16.4 optimum==1.12.0 -safetensors==0.3.2 +safetensors==0.3.3 Flask==2.2.3 Flask-SocketIO==5.3.2 python-socketio==5.7.2