Hugging Face's logo

Spaces:
jkorstad
/
Mesh_Rigger
Running on Zero

App Files Community
jkorstad commited on
Commit
7eda35c
·
verified ·
1 Parent(s): 9d43573

Update setup_blender.sh

Browse files
Files changed (1) hide show
  1. setup_blender.sh +2 -2
setup_blender.sh CHANGED
@@ -24,8 +24,8 @@ TARGET_CUDA_VERSION_SHORT="cu121" # PyTorch 2.3.1 wheels are available for cu121
24
  TORCH_INDEX_URL="https://download.pytorch.org/whl/${TARGET_CUDA_VERSION_SHORT}"
25
 
26
  # Direct URL for the compatible flash-attn wheel for v2.5.8
27
- # Corresponds to Python 3.11 (cp311), PyTorch 2.3.1+cu121
28
- FLASH_ATTN_WHEEL_URL="https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+pt23cu121-cp311-cp311-linux_x86_64.whl"
29
 
30
  # --- Set Environment Variables for Build ---
31
  export CUDA_HOME=${CUDA_HOME:-/usr/local/cuda} # This might be nominal if nvcc isn't actually used
 
24
  TORCH_INDEX_URL="https://download.pytorch.org/whl/${TARGET_CUDA_VERSION_SHORT}"
25
 
26
  # Direct URL for the compatible flash-attn wheel for v2.5.8
27
+ # Compatible with: Python 3.11 (cp311), PyTorch 2.3.x (torch2.3), CUDA 11.8 (cu118), CXX11 ABI TRUE
28
+ FLASH_ATTN_WHEEL_URL="https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu118torch2.3cxx11abiTRUE-cp311-cp311-linux_x86_64.whl"
29
 
30
  # --- Set Environment Variables for Build ---
31
  export CUDA_HOME=${CUDA_HOME:-/usr/local/cuda} # This might be nominal if nvcc isn't actually used