model = GPT2LMHeadModel.from_pretrained(
    "gpt2",
    load_in_8bit=True,
    device_map="auto"
)