LICENSE
README.md
pyproject.toml
simple_stories_train/__init__.py
simple_stories_train/dataloaders.py
simple_stories_train/log.py
simple_stories_train/tinyshakespeare.py
simple_stories_train/tokenizer.py
simple_stories_train/train_gpt2.py
simple_stories_train/train_llama.py
simple_stories_train/utils.py
simple_stories_train.egg-info/PKG-INFO
simple_stories_train.egg-info/SOURCES.txt
simple_stories_train.egg-info/dependency_links.txt
simple_stories_train.egg-info/requires.txt
simple_stories_train.egg-info/top_level.txt
simple_stories_train/models/llama.py
simple_stories_train/models/model_configs.py
tests/test_dataloader.py
tests/test_llama_implementation.py
tests/test_utils.py