How to use MFA?

1) How to use Metal Flash Attention with the UNet model? 2) Also, is there any way I could only load the 6bit weights in the memory rather than 16bit?