debug

Signed-off-by: Mehant Kammakomati <[email protected]>

Author: kmehant

tox.ini

@@ -42,7 +42,7 @@ commands =
     python -m fms_acceleration.cli install -e {toxinidir}/plugins/accelerated-moe
 
     # install the flash attn at the last 
-    pip install flash-attn
+    pip install flash-attn --no-build-isolation
 
     # run the benchmark script
     bash scripts/run_benchmarks.sh {posargs:"1 2" "4 8" benchmark_outputs}