How to load model after train?

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.11k stars 101 forks source link

How to load model after train? #19

Closed ntdat02092002 closed 1 year ago

ntdat02092002 commented 1 year ago

After training using your training code example, I have file .bin. Can you tell me how to load it to use generate level? Thank you.