Welcome to the Tiny World Model project! This project implements a simplified world model designed for educational and research purposes. The model simulates a basic environment where we get a causal language model to predict future images. This README will guide you through the setup, usage, and development process of the project.
All relevant commands can be found in the makefile. Run make help to see them.
It is designed to be run inside a docker container and these experiments were run using lambda cloud so should be easy to recreate there.
To start the training simply run make train
This project is licensed under the MIT License. See the LICENSE file for details.
Thank you for using the Tiny World Model! We hope this project helps you in your research and learning journey. If you have any questions, feel free to reach out. Happy coding!