Commit 14a79a12 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Attempt to fix FlashAttention installation error

parent ef239f30
...@@ -15,7 +15,7 @@ wget -P /tmp \ ...@@ -15,7 +15,7 @@ wget -P /tmp \
export PATH=lib/conda/bin:$PATH export PATH=lib/conda/bin:$PATH
lib/conda/bin/python3 -m pip install nvidia-pyindex lib/conda/bin/python3 -m pip install nvidia-pyindex
conda env create --name=${ENV_NAME} -f environment.yml conda env create --name=${ENV_NAME} -f environment.yml
source activate ${ENV_NAME} source scripts/activate_conda_env.sh
echo "Attempting to install FlashAttention" echo "Attempting to install FlashAttention"
pip install git+https://github.com/HazyResearch/flash-attention.git@5b838a8bef78186196244a4156ec35bbb58c337d && echo "Installation successful" pip install git+https://github.com/HazyResearch/flash-attention.git@5b838a8bef78186196244a4156ec35bbb58c337d && echo "Installation successful"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment