alexnasa/

flash-attn-3

This repo provides the wheel for Flash Attention 3 for Hopper architecture for the current default cuda version on ZeroGPU spaces, which at the moment is Torch version: 2.8.0+cu128.

You can add this to the requirements.txt file of your space