Skip to content

Commit

Permalink
fix: remove GO_FAST if support is not present (#347)
Browse files Browse the repository at this point in the history
If we want to test whether this works on new systems, it should clean up after itself if the overwride is no longer present.
  • Loading branch information
HPPinata authored Nov 18, 2024
1 parent 6dcfce3 commit c395b9d
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions horde_worker_regen/amd_go_fast/install_amd_go_fast.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#!/bin/bash

if [ "${FLASH_ATTENTION_USE_TRITON_ROCM^^}" == "TRUE" ]; then
if ! pip install -U pytest git+https://github.com/ROCm/flash-attention@micmelesse/upstream_pr_rebase; then
if ! pip install -U pytest git+https://github.com/ROCm/flash-attention@micmelesse/upstream_pr_rebase; then
echo "Tried to install flash attention and failed!"
else
echo "Installed flash attn."
Expand All @@ -11,7 +11,9 @@ if [ "${FLASH_ATTENTION_USE_TRITON_ROCM^^}" == "TRUE" ]; then
else
echo "Installed AMD GO FAST."
fi
fi
fi
else
echo "Did not detect support for AMD GO FAST"
echo "Did not detect support for AMD GO FAST. Cleaning up."
pip uninstall flash_attn
rm "${PY_SITE_DIR}"/hordelib/nodes/amd_go_fast.py
fi

0 comments on commit c395b9d

Please sign in to comment.