config.json

{
    "architectures": ["CustomModel"],
    "task": "feature-extraction"
  }