How to use MarioBert?

[kownse]

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]

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]

Thank you @shyamsn97 , looking forward to it.

[Ezio77]

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