return flash attention

Author: James Scott

.github/workflows/ci.yml

@@ -27,7 +27,7 @@ jobs:
     - name: Setup Environment
       uses: prefix-dev/[email protected]
       with:
-        pixi-version: v0.47.0
+        pixi-version: v0.54.0
         cache: false
         cache-write: ${{ github.event_name == 'push' && github.ref_name == 'main' }}
 
@@ -45,7 +45,7 @@ jobs:
     - name: Setup Environment
       uses: prefix-dev/[email protected]
       with:
-        pixi-version: v0.47.0
+        pixi-version: v0.54.0
         cache: false
         cache-write: ${{ github.event_name == 'push' && github.ref_name == 'main' }}