Fix requirements.txt build order

requirements.txt

@@ -1,9 +1,10 @@
---extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.2.2
-torchaudio==2.2.2
-torchvision==0.17.2
+# Step 1: Install PyTorch and its ecosystem first
+--find-links https://download.pytorch.org/whl/torch_stable.html
+torch==2.2.2+cu121
+torchaudio==2.2.2+cu121
+torchvision==0.17.2+cu121
 
-# From conda environment
+# Step 2: Install other dependencies
 absl-py==2.2.2
 accelerate==1.6.0
 addict==2.4.0