config.json

{
  "model_type": "custom",
  "library_name": "pytorch"
}