Update requirements.txt

requirements.txt

@@ -1,8 +1,10 @@
-torch>=2.2
-transformers>=4.40
-diffusers @ git+https://github.com/huggingface/diffusers.git
-accelerate>=0.27
-safetensors
-gradio>=4.22
-numpy
-pillow
+# requirements.txt
+ torch>=2.2
++torchvision>=0.17          # 0.17.* pairs with torch-2.2.*
+ transformers>=4.40
+ diffusers @ git+https://github.com/huggingface/diffusers.git
+ accelerate>=0.27
+ safetensors
+ gradio>=4.22
+ numpy
+ pillow