philschmid's picture
philschmid HF staff
Create gpt2.json
3bdb891 verified
raw
history blame
278 Bytes
{
"openai-community/gpt2": [
{
"batch_size": 1,
"sequence_length": 1024,
"num_cores": 1,
"auto_cast_type": "fp16"
},
{
"batch_size": 16,
"sequence_length": 1024,
"num_cores": 1,
"auto_cast_type": "fp16"
}
]
}