move PY_SITE_DIR

Haidra-Org · Dec 17, 2024 · b59a3a4 · b59a3a4
1 parent 95d3637
commit b59a3a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/horde_worker_regen/amd_go_fast/install_amd_go_fast.sh b/horde_worker_regen/amd_go_fast/install_amd_go_fast.sh
@@ -1,11 +1,11 @@
 #!/bin/bash
+PY_SITE_DIR=$(python -c "import sysconfig; print(sysconfig.get_path('purelib'))")
 
 if [ "${FLASH_ATTENTION_TRITON_AMD_ENABLE^^}" == "TRUE" ]; then
 	if ! pip install -U pytest git+https://github.com/Dao-AILab/flash-attention; then
 		echo "Tried to install flash attention and failed!"
 	else
 		echo "Installed flash attn."
-		PY_SITE_DIR=$(python -c "import sysconfig; print(sysconfig.get_path('purelib'))")
 		if ! cp horde_worker_regen/amd_go_fast/amd_go_fast.py "${PY_SITE_DIR}"/hordelib/nodes/; then
 			echo "Failed to install AMD GO FAST."
 		else