diff --git a/requirements.txt b/requirements.txt index bb3fd1004..f2df922e0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ torch torchsde einops transformers>=4.25.1 -safetensors>=0.3.2 +safetensors==0.3.2 aiohttp accelerate pyyaml