torch.manual_seed(123)

token_ids = generate(
    model=gpt,
    idx=text_to_token_ids(
        "today's hit movies are",
        tokenizer
    ).to(device),
    max_new_tokens=50,
    context_size=NEW_CONFIG["context_length"],
    top_k=50,
    temperature=1.5
)

print(
    "Output text:",
    token_ids_to_text(token_ids, tokenizer)
)

        __
        __