Flash Attention Installation Error

[LaurinLang]

Using the current environment.yml file with mamba results in the following errors.
Looks like the hash for the flash attention version might not be current?

Pip subprocess output:
Collecting git+https://github.com/NVIDIA/dllogger.git (from -r /home/lang/openfold/condaenv.81prc7je.requirements.txt (line 3))
Cloning https://github.com/NVIDIA/dllogger.git to /tmp/pip-req-build-x5ixqy35
Resolved https://github.com/NVIDIA/dllogger.git to commit 0540a43971f4a8a16693a9de9de73c1072020769
Preparing metadata (setup.py): started
Preparing metadata (setup.py): finished with status 'done'
Collecting git+https://github.com/Dao-AILab/flash-attention.git@5b838a8 (from -r /home/lang/openfold/condaenv.81prc7je.requirements.txt (line 4))
Cloning https://github.com/Dao-AILab/flash-attention.git (to revision 5b838a8) to /tmp/pip-req-build-fdzk8ub6
Resolved https://github.com/Dao-AILab/flash-attention.git to commit 5b838a8
Preparing metadata (setup.py): started
Preparing metadata (setup.py): finished with status 'error'

Pip subprocess error:
Running command git clone --filter=blob:none --quiet https://github.com/NVIDIA/dllogger.git /tmp/pip-req-build-x5ixqy35
Running command git clone --filter=blob:none --quiet https://github.com/Dao-AILab/flash-attention.git /tmp/pip-req-build-fdzk8ub6
WARNING: Did not find branch or tag '5b838a8', assuming revision or ref.
Running command git checkout -q 5b838a8
Running command git submodule update --init --recursive -q
error: subprocess-exited-with-error

[xuagu37]

Same error here. Any solutions please?

[lonelyhwy]

same error. Do you have any solutions?

[abhinavb22]

I have the same error. It works fine in my PC but I get this error when I try to install in my HPC facility. Were you able to fix this issue ?