Update requirements.txt

requirements.txt

@@ -1,6 +1,24 @@
-accelerate
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+# Core dependencies
+torch==2.4.0
+torchvision==0.19.0
+pillow==10.4.0
+tqdm==4.67.1
+gradio==4.44.0
+spaces
+
+# Diffusion and model dependencies
+transformers==4.46.3
 diffusers
-invisible_watermark
-torch
-transformers
-xformers
+accelerate
+xformers==0.0.27.post2
+
+# Quantization and optimization
+bitsandbytes
+gguf
+peft
+sentencepiece
+
+# Optional but recommended for faster inference
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl