Add --no_balance flag to not balance datasets#287

AlexTMallen wants to merge 4 commits intomainEleutherAI/elk:mainfrom

no-balanceEleutherAI/elk:no-balanceCopy head branch name to clipboard