model = GPTModel(GPT_CONFIG_124M)

model.load_state_dict(
    torch.load("model.pth", map_location=device)
)

model.eval()
print(model)

        __
        __