Update setup_blender.sh

setup_blender.sh

@@ -20,7 +20,8 @@ UNIRIG_REPO_CLONE_DIR="${APP_DIR}/UniRig"
 
 TORCH_VERSION="2.3.1"
 TORCHVISION_VERSION="0.18.1"
-TARGET_CUDA_VERSION_SHORT="cu121" # PyTorch 2.3.1 wheels are available for cu121
+# ** MODIFIED: Changed PyTorch target to CUDA 11.8 to align with flash-attn wheel **
+TARGET_CUDA_VERSION_SHORT="cu118"
 TORCH_INDEX_URL="https://download.pytorch.org/whl/${TARGET_CUDA_VERSION_SHORT}"
 
 # Direct URL for the compatible flash-attn wheel for v2.5.8
@@ -38,7 +39,7 @@ export C_INCLUDE_PATH="${PYTHON_INCLUDE_DIR}:${C_INCLUDE_PATH}"
 export CPLUS_INCLUDE_PATH="${PYTHON_INCLUDE_DIR}:${CPLUS_INCLUDE_PATH}"
 
 # TORCH_CUDA_ARCH_LIST is important if flash-attn *does* try to compile parts of itself.
-export TORCH_CUDA_ARCH_LIST="8.0;8.6;9.0" # Common GPU architectures
+export TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;9.0" # Added older architectures for cu118 compatibility
 
 echo "--- Setup Script Start ---"
 echo "Target Blender Installation Directory: ${INSTALL_DIR}"
@@ -118,6 +119,8 @@ echo "flash-attn installation attempted from wheel."
 echo "Step 3: Installing remaining dependencies from ${UNIRIG_REQS_FILE_IN_SPACE}..."
 # Ensure flash-attn is REMOVED from unirig_requirements.txt.
 # This will install torch-scatter, torch-cluster, spconv, bpy, etc.
+# PyG (torch-scatter, etc.) links in unirig_requirements.txt might need to be updated for torch 2.3 + cu118
+# Example: torch-scatter -f https://data.pyg.org/whl/torch-2.3.0+cu118.html (adjust torch version if needed)
 "${BLENDER_PY_EXEC}" -m pip install --no-cache-dir \
     -r "${UNIRIG_REQS_FILE_IN_SPACE}" -vvv