1 files changed, 4 insertions, 0 deletions

diff --git a/requirements.txt b/requirements.txt
index 0fba0b23..05818aa6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+accelerate

 basicsr

 diffusers

 fairscale==0.4.4

@@ -27,3 +28,6 @@ kornia
 lark

 inflection

 GitPython

+torchsde

+safetensors

+psutil; sys_platform == 'darwin'