Update setup_blender.sh

setup_blender.sh

@@ -20,8 +20,8 @@ UNIRIG_REPO_CLONE_DIR="${APP_DIR}/UniRig"
 
 TORCH_VERSION="2.3.1"
 TORCHVISION_VERSION="0.18.1"
-# ** MODIFIED: Changed to cu122 to align with common flash-attn pre-built wheels **
-TARGET_CUDA_VERSION_SHORT="cu122"
+# ** MODIFIED: Reverted to cu121 as PyTorch 2.3.1 wheels are readily available for it **
+TARGET_CUDA_VERSION_SHORT="cu121"
 TORCH_INDEX_URL="https://download.pytorch.org/whl/${TARGET_CUDA_VERSION_SHORT}"
 
 # --- Set Environment Variables for Build ---
@@ -117,7 +117,7 @@ echo "Step 2: (flash-attn will be installed via requirements file in the next st
 echo "Step 3: Installing remaining dependencies from ${UNIRIG_REQS_FILE_IN_SPACE}..."
 # This will install torch-scatter, torch-cluster, spconv, flash-attn (if listed and compatible wheel found), bpy, etc.
 # flash-attn build (if a wheel is not found) would use TORCH_CUDA_ARCH_LIST.
-# Hopefully, a pre-built wheel for flash-attn (e.g., +pt23cu122) will be found due to PyTorch being cu122.
+# Hopefully, a pre-built wheel for flash-attn (e.g., +pt23cu121) will be found due to PyTorch being cu121.
 "${BLENDER_PY_EXEC}" -m pip install --no-cache-dir -r "${UNIRIG_REQS_FILE_IN_SPACE}" -vvv
 
 echo "Dependency installation for Blender's Python complete."