INNER CODE UNIT · Python

run

HendrikStrobelt/detecting-fake-text · preload_gpt2.py:5

def run(model_name_or_path):

    GPT2Tokenizer.from_pretrained(model_name_or_path)
    GPT2LMHeadModel.from_pretrained(model_name_or_path)
    print("Loaded GPT-2 model!")


if __name__ == '__main__':
    run("gpt2")

View source record →