config.json

{
  "_name_or_path": "StanfordNeuroAILab/PSI",
  "architectures": ["PSI"],
  "auto_map": {
    "AutoConfig": "config.PSIConfig",
    "AutoModel": "psi.PSI"
  },
  "model_type": "PSI"
}