7cpc / python
1c1's picture
Create python
63ae938 verified
raw
history blame contribute delete
99 Bytes
model = GPT2LMHeadModel.from_pretrained(
"gpt2",
load_in_8bit=True,
device_map="auto"
)