1 files changed, 4 insertions, 1 deletions
diff --git a/requirements.txt b/requirements.txt
index 0fba0b23..678acb4d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
+accelerate
 basicsr
-diffusers
 fairscale==0.4.4
 fonts
 font-roboto
@@ -27,3 +27,6 @@ kornia
 lark
 inflection
 GitPython
+torchsde
+safetensors
+psutil; sys_platform == 'darwin'