shyamsn97/mario-gpt

How to use MarioBert?

kownse opened this issue · 3 comments

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'

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 :)

Thank you @shyamsn97 , looking forward to it.

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