Upload requirements.txt

requirements.txt

@@ -1,22 +1,35 @@
-gradio==5.35.0
-spaces==0.37.1
-torch==2.3.1
-torchaudio==2.3.1
-torchvision==0.18.1
-transformers==4.44.2
-Pillow==10.1.0
-sentencepiece==0.2.0
-vector-quantize-pytorch==1.18.5
-vocos==0.1.0
-accelerate==1.2.1
-timm==0.9.10
-soundfile==0.12.1
-librosa==0.9.0
-decord
-moviepy
-numpy>=1.24.0
-huggingface_hub>=0.33.2
-protobuf==3.20.3
-einops>=0.8.0
-einx>=0.3.0
-flash-attn
+# Core PyTorch dependencies - install first
+torch==2.3.1
+torchaudio==2.3.1
+torchvision==0.18.1
+
+# Flash Attention - needs torch to be installed first
+flash-attn==2.8.0.post2
+
+# Core ML libraries
+transformers==4.44.2
+accelerate==1.2.1
+numpy>=1.24.0
+einops>=0.8.0
+einx>=0.3.0
+
+# Gradio and Spaces
+gradio==5.35.0
+spaces==0.37.1
+
+# Image and video processing
+Pillow==10.1.0
+decord
+moviepy
+
+# Audio processing
+soundfile==0.12.1
+librosa==0.9.0
+
+# Additional ML dependencies
+sentencepiece==0.2.0
+vector-quantize-pytorch==1.18.5
+vocos==0.1.0
+timm==0.9.10
+huggingface_hub>=0.33.2
+protobuf==3.20.3