Lint flash_attn.py

config chooser, update readme instructions, device config, llama flash attention, debug out the labels, fix config key checks, other bugfixes