pip install mario-gpt

git clone git@github.com:shyamsn97/mario-gpt.git
python setup.py install

from mario_gpt import MarioLM, SampleOutput
# pretrained_model = shyamsn97/Mario-GPT2-700-context-length
mario_lm = MarioLM()
...

import torch
from mario_gpt import MarioDataset, MarioLM, TrainingConfig, MarioGPTTrainer
# 기본 gpt 모델 생성
...

from mario_gpt import MarioLM
mario_lm = MarioLM()
prompts = [

- 기본 추론 코드 (Basic inference code)
- MarioBert 모델 추가 (Add MarioBert Model)
- 인터랙티브 시뮬레이터 추가 (Add Interactive simulator)
- 논문 기반 학습 코드 (Training code from paper)
- 논문 기반 인페인팅 기능 (Inpainting functionality from paper)
- 개방형 레벨 생성 코드 (Open-ended level generation code)
- 다양한 생성 방법 (예: 제약된 빔 검색 등) (Different generation methods (eg. constrained beam search, etc.))

Shyam Sudhakaran shyamsnair@protonmail.com, https://github.com/shyamsn97, https://shyamsn97.github.io/

Miguel González-Duque migd@itu.dk, https://github.com/miguelgondu

Claire Glanois clgl@itu.dk, https://github.com/claireaoi

Matthias Freiberger matfr@itu.dk, https://github.com/matfrei

Elias Najarro enaj@itu.dk, https://github.com/enajx

Sebastian Risi sebr@itu.dk, https://github.com/sebastianrisi, https://sebastianrisi.com/

학술적 또는 상업적 용도로 코드를 사용하시는 경우, 관련 논문을 인용해 주십시오: