Update setup_blender.sh

setup_blender.sh

@@ -23,9 +23,6 @@ TORCHVISION_VERSION="0.18.1"
 TARGET_CUDA_VERSION_SHORT="cu121" # PyTorch 2.3.1 wheels are available for cu121
 TORCH_INDEX_URL="https://download.pytorch.org/whl/${TARGET_CUDA_VERSION_SHORT}"
 
-# flash-attn version known to have compatible wheels (Python 3.11, PyTorch 2.3.1+cu121)
-FLASH_ATTN_VERSION="2.5.8"
-
 # --- Set Environment Variables for Build ---
 export CUDA_HOME=${CUDA_HOME:-/usr/local/cuda}
 export PATH="${CUDA_HOME}/bin:${LOCAL_BIN_DIR}:${PATH}"
@@ -36,7 +33,8 @@ export CPATH="${PYTHON_INCLUDE_DIR}:${CPATH}"
 export C_INCLUDE_PATH="${PYTHON_INCLUDE_DIR}:${C_INCLUDE_PATH}"
 export CPLUS_INCLUDE_PATH="${PYTHON_INCLUDE_DIR}:${CPLUS_INCLUDE_PATH}"
 
-export TORCH_CUDA_ARCH_LIST="8.0;8.6;9.0" # General set for compatibility
+# TORCH_CUDA_ARCH_LIST might still be useful for other compiled packages if any
+export TORCH_CUDA_ARCH_LIST="8.0;8.6;9.0"
 
 echo "--- Setup Script Start ---"
 echo "Target Blender Installation Directory: ${INSTALL_DIR}"
@@ -44,7 +42,7 @@ echo "Blender Python Executable: ${BLENDER_PY_EXEC}"
 echo "Using CUDA_HOME=${CUDA_HOME}"
 echo "Targeting PyTorch for CUDA: ${TARGET_CUDA_VERSION_SHORT}"
 echo "TORCH_CUDA_ARCH_LIST: ${TORCH_CUDA_ARCH_LIST}"
-echo "Attempting to install flash-attn version: ${FLASH_ATTN_VERSION}"
+echo "NOTE: flash-attn installation is now SKIPPED."
 
 # --- Download and Extract Blender ---
 mkdir -p "${BLENDER_INSTALL_BASE}"
@@ -102,18 +100,11 @@ echo "Step 1: Installing PyTorch ${TORCH_VERSION} (for CUDA ${TARGET_CUDA_VERSIO
     --index-url ${TORCH_INDEX_URL} -vvv
 echo "PyTorch and Torchvision installation attempted."
 
-echo "Step 2: Installing flash-attn==${FLASH_ATTN_VERSION} from PyPI (binary only)..."
-# Install flash-attn from PyPI, specifying version and forcing binary only
-# The --only-binary flag applies to the package name immediately following it.
-"${BLENDER_PY_EXEC}" -m pip install --no-cache-dir \
-    --only-binary flash-attn \
-    flash-attn==${FLASH_ATTN_VERSION} -vvv
-echo "flash-attn installation attempted from PyPI."
+# Step 2 for flash-attn has been removed.
 
-echo "Step 3: Installing remaining dependencies from ${UNIRIG_REQS_FILE_IN_SPACE}..."
-# Ensure flash-attn is REMOVED from unirig_requirements.txt to avoid conflicts or reinstallation attempts.
+echo "Step 2 (formerly 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.
-# These packages should find the PyTorch and flash-attn already installed.
 "${BLENDER_PY_EXEC}" -m pip install --no-cache-dir \
     -r "${UNIRIG_REQS_FILE_IN_SPACE}" -vvv