diff --git a/requirements.txt b/requirements.txt
index bead3f90..9e97eed3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 accelerate==0.33.0
 transformers==4.44.0
-diffusers==0.25.0
+diffusers[torch]==0.25.0
 deepspeed==0.16.7
 ftfy==6.1.1
 # albumentations==1.3.0