run nightly ci builds against upstream main (#1851)

* run nightly ci builds against upstream main * add test badges * run the multigpu tests against nightly main builds too
2024-08-22 13:10:54 -04:00
parent 2f8037fee6
commit dcbff16983
6 changed files with 137 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -80,7 +80,7 @@ setup(
    dependency_links=dependency_links,
    extras_require={
        "flash-attn": [
-            "flash-attn==2.6.2",
+            "flash-attn==2.6.3",
        ],
        "fused-dense-lib": [
            "fused-dense-lib  @ git+https://github.com/Dao-AILab/flash-attention@v2.6.2#subdirectory=csrc/fused_dense_lib",