Added Sampling class, Simulator class, and better saving functionalities

shyamsn97 / mario-gpt

[Neurips 2023] Generating Mario Levels with GPT2. Code for the paper "MarioGPT: Open-Ended Text2Level Generation through Large Language Models" https://arxiv.org/abs/2302.05981

https://huggingface.co/shyamsn97/Mario-GPT2-700-context-length

MIT License

1.12k stars 101 forks source link

Added Sampling class, Simulator class, and better saving functionalities #9

Closed shyamsn97 closed 1 year ago

shyamsn97 commented 1 year ago

Refactored LM sampling into a new class
SampleOutput dataclass has convenient functions
- generating images
- generating text representations
- playing the level interactively (using the new Simulator class)
- running an Astar agent in the level (using the new Simulator class)