Does it support a stable diffusion model

Dao-AILab / flash-attention

Fast and memory-efficient exact attention

BSD 3-Clause "New" or "Revised" License

13.26k stars 1.2k forks source link

Open libai-lab opened 1 month ago

libai-lab commented 1 month ago

Does it support a stable diffusion model

tridao commented 1 month ago

If stable diffusion uses attention, then yes