improve prepared dataset loading, fix inference

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