Skip to content

GitLab

Commit 7384e2d6 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Tweak install script

parent feed4ae2
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 1 deletion
scripts/install_third_party_dependencies.sh
View file @ 7384e2d6
......@@ -18,7 +18,12 @@ conda env create --name=${ENV_NAME} -f environment.yml
source scripts/activate_conda_env.sh
echo "Attempting to install FlashAttention"
pip install git+https://github.com/HazyResearch/flash-attention.git@5b838a8bef78186196244a4156ec35bbb58c337d && echo "Installation successful"
git clone https://github.com/HazyResearch/flash-attention
CUR_DIR=$PWD
cd flash-attention
git checkout 5b838a8bef
python3 setup.py install
cd $CUR_DIR
# Install DeepMind's OpenMM patch
OPENFOLD_DIR=$PWD
......
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