namer / config.json
Edwin Jose Palathinkal
Fix HuggingFace API compatibility
4b0dbc8
{
"architectures": [
"NamerModel"
],
"auto_map": {
"AutoModel": "modeling_namer.NamerModel"
},
"d_model": 128,
"dim_feedforward": 512,
"dropout": 0.0,
"dtype": "float32",
"eos_token_id": 40,
"max_output_len": 35,
"model_type": "custom",
"nhead": 4,
"num_encoder_layers": 4,
"pad_token_id": 10,
"transformers_version": "5.8.0",
"vocab_size": 41
}