TransHLA_II / config.json
SkywalkerLu's picture
Upload model
29c8230 verified
{
"architectures": [
"TransHLA_II_Model"
],
"auto_map": {
"AutoConfig": "configuration_TransHLA_II.TransHLA_II_Config",
"AutoModel": "modeling_TransHLA_II.TransHLA_II_Model"
},
"cnn_kernel_size": 3,
"cnn_num_channel": 256,
"cnn_padding_index": 0,
"cnn_padding_size": 1,
"cnn_stride": 1,
"d_ff": 64,
"d_model": 1280,
"max_len": 21,
"model_type": "TransHLA",
"n_head": 8,
"n_layers": 6,
"pooling_size": 2,
"region_embedding_size": 3,
"torch_dtype": "float32",
"transformers_version": "4.24.0"
}