diff --git a/setup_files/requirements.txt b/setup_files/requirements.txt
index 538b323..8c5f7dd 100644
--- a/setup_files/requirements.txt
+++ b/setup_files/requirements.txt
@@ -2,7 +2,7 @@ invisible_watermark
 accelerate
 diffusers[torch]==0.21.4
 onnxruntime==1.16.1
-safetensors==0.3.3
+safetensors==0.4.0
 torch==2.0.1+cu117
 transformers==4.34.1
 xformers==0.0.22