improve prepared dataset loading, fix inference

helpful info output

various bugfixes

more fixes and prep for llama training

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

make it work with pythia in the cloud

WIP for axolotl trainer