[Enhancement] Any plan or advice on adding `flash_attn` support?

Hey, I am astonished by the perfect performance provided by ChatterBox voice cloning model.

Except for the relatively slow inference speed.

Do you have any plans to add flash_attn support?

Or, if I would like to modify the code to add Flash Attention support, is there any suggestions to provide?