Why this mario gpt script is laggy?

[DrFrankatron]

This is the script:

from mario_gpt import MarioLM

mario_lm = MarioLM()

prompts = ["many pipes, many enemies, some blocks, high elevation"]

generated_level = mario_lm.sample(
prompts=prompts,
num_steps=100,
temperature=2.0,
use_tqdm=True
)

play in interactive

generated_level.play()

run Astar agent

generated_level.run_astar()

And I need more num_steps, but the AI lags when it reach more than 100 steps... its possible to get performance with this python library?

[DrFrankatron]

This is my code, sorry the bad formatted previous one:

from mario_gpt import MarioLM

mario_lm = MarioLM()

prompts = ["many pipes, many enemies, some blocks, high elevation"]

generated_level = mario_lm.sample(
prompts=prompts,
num_steps=100,
temperature=2.0,
use_tqdm=True
)

generated_level.play()

generated_level.run_astar()

[shyamsn97]

Hey! Are you sure you’re using a gpu? It can be pretty slow without one. Basically has the steps increase the information in the context increases.

[shyamsn97]

You can check with torch.cuda.is_available()