SimpleStories is a more diverse version: https://huggingface.co/datasets/SimpleStories/SimpleStories