Update unirig_requirements.txt

unirig_requirements.txt

@@ -8,12 +8,14 @@ numpy==1.26.4
 # Example using index-url, adjust if needed
 # Note: This will be installed *first* by setup_blender.sh, but listing it here ensures
 # pip knows the version constraint when resolving other dependencies later.
+
+#flash-attn pre-built whl
+flash-attn==2.5.9.post1 
+
+# Make sure torch, torchvision, torch-scatter, etc. are also listed with their correct versions/flags
 torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121
 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121
 
-# PyTorch Geometric libraries (for CUDA 12.1, Python 3.11)
-# Use -f to point pip to the HTML index pages. Pip should find the correct wheels.
-# Make sure these URLs correspond to torch 2.3.1 + cu121
 torch-scatter -f https://data.pyg.org/whl/torch-2.3.1+cu121.html
 torch-cluster -f https://data.pyg.org/whl/torch-2.3.1+cu121.html
 
@@ -30,7 +32,6 @@ lightning         # Might be redundant if pytorch_lightning is used
 addict
 timm
 fast-simplification
-flash-attn==2.5.9.post1
 trimesh
 open3d
 pyrender