Update unirig_requirements.txt

unirig_requirements.txt

@@ -1,22 +1,25 @@
 # UniRig Dependencies (to be installed into Blender's Python 3.11)
 
+# Numpy (UniRig: 1.26.4)
+numpy==1.26.4
+
 # PyTorch (Matching main env for simplicity, ensure compatibility with UniRig >=2.3.1)
 # Install the version compatible with Python 3.11 and CUDA 12.1
 # 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.
 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)
-# Note the updated URLs for cp311
-torch-scatter -f https://data.pyg.org/whl/torch-2.3.1+cu121/torch_scatter-latest+pt23cu121-cp311-cp311-linux_x86_64.html
-torch-cluster -f https://data.pyg.org/whl/torch-2.3.1+cu121/torch_cluster-latest+pt23cu121-cp311-cp311-linux_x86_64.html
+# 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
 
 # Spconv (for CUDA 12.1 - check compatibility with Python 3.11 if issues arise)
 spconv-cu121
 
-# Numpy (UniRig: 1.26.4)
-numpy==1.26.4
-
 # Dependencies from UniRig's official requirements.txt (excluding bpy and flash_attn)
 transformers
 python-box