dacorvo's picture
dacorvo HF staff
Add more gpt2 configurations
3fbf810 verified
raw
history blame
398 Bytes
{
"openai-community/gpt2": [
{
"batch_size": 1,
"sequence_length": 1024,
"num_cores": 2,
"auto_cast_type": "fp16"
},
{
"batch_size": 4,
"sequence_length": 1024,
"num_cores": 2,
"auto_cast_type": "fp16"
},
{
"batch_size": 16,
"sequence_length": 1024,
"num_cores": 2,
"auto_cast_type": "fp16"
}
]
}