search

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 103 forks source link

How to use MarioBert? #15

Open kownse opened 1 year ago

kownse commented 1 year ago

May I ask how to use the MarioBert in this repo? I tried to call MarioLM(mask_model=True), but MarioBert has no sample method:

    generated_level = mario_lm.sample(
AttributeError: 'MarioBert' object has no attribute 'sample'
shyamsn97 commented 1 year ago

Hey! Currently this is used for the inpainting functionality used in the novelty search portion of the work. I'm working on cleaning up that code and releasing it soon :)

kownse commented 1 year ago

Thank you @shyamsn97 , looking forward to it.

Ezio77 commented 4 months ago

Have you implemented the relevant code for MarioBert? I tried to complete it myself but failed.