diff --git a/requirements.txt b/requirements.txt
index 7539e55..78c6303 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,8 @@ lightning==2.5.0.post0
 omegaconf==2.3.0
 torch==2.5.0
 diffusers==0.30.0
-jsonargparse[signatures]>=4.27.7
\ No newline at end of file
+jsonargparse[signatures]>=4.27.7
+torchvision
+timm
+accelerate
+gradio
\ No newline at end of file