* flash attn changes
* allow specifying specific xformers op
* make optional xformers import
* change attention variant to be enum