refactor trainer setup to account for deepspeed integration

improve prepared dataset loading, fix inference

helpful info output

fix issue with completed model being empty

various bugfixes

bettter handling of llama model import

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

black formatting

make it work with pythia in the cloud

WIP for axolotl trainer