ydshieh
/

pt_custom_model_example

Model card Files Files and versions Community

File size: 179 Bytes

{
  "n_layers": 2,
  "model_type": "my_model",
  "auto_map": {
    "AutoConfig": "configuration_my_model.MyModelConfig",
    "AutoModel": "modeling_my_model.MyMode"
  }
}