config.json

{
  "model_type": "npk",
  "architectures": ["NPKPredictionModel"],
  "auto_map": {
    "AutoConfig": "modeling_npk.NPKConfig",
    "AutoModel": "modeling_npk.NPKPredictionModel"
  },
  "trust_remote_code": true
}