Various dataloader updates and fixes#144

daviswer wants to merge 15 commits intomainfoundation-model-stack/fms-fsdp:mainfrom

data-fixesfoundation-model-stack/fms-fsdp:data-fixesCopy head branch name to clipboard