Flash Attention 2.8.1

I lost a lot of time to install all dependencies, but i'm stuck with flash attention because the wrong version:

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/home/ryzen/FramePack/magcache_demo_gradio_f1.py", line 17, in
from diffusers import AutoencoderKLHunyuanVideo
File "", line 1075, in _handle_fromlist
File "/home/ryzen/miniconda3/envs/framepack/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 811, in getattr
value = getattr(module, name)
File "/home/ryzen/miniconda3/envs/framepack/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 810, in getattr
module = self._get_module(self._class_to_module[name])
File "/home/ryzen/miniconda3/envs/framepack/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 822, in _get_module
raise RuntimeError(
RuntimeError: Failed to import diffusers.models.autoencoders.autoencoder_kl_hunyuan_video because of the following error (look up to see its traceback):
Requires Flash-Attention version >=2.7.1,<=2.8.0 but got 2.8.1.

The new version of flash attention is not supported by FramePack... Is there a solution or i have to reinstall everything again? I'm on linux in this machine.

Start with magcache_demo_gradio_f1.py and the import of AutoencoderKLHunyuanVideo shown in the traceback. Check how FramePack declares or validates the Flash-Attention dependency and reproduce the import failure with version 2.8.1. Done means the supported dependency path is clear and the import no longer fails.