AMPLIFY_benchmark / config.json
Lolalb's picture
Upload AMPLIFY
4ba3699 verified
{
"_name_": "AMPLIFY",
"ambiguous_token_ids": [
1,
6,
7,
8,
9,
10,
11
],
"architectures": [
"AMPLIFY"
],
"auto_map": {
"AutoConfig": "amplify.AMPLIFYConfig",
"AutoModel": "amplify.AMPLIFY"
},
"base_scale": 0.03227486121839514,
"bos_token_id": 3,
"decoder_init_range": 0.02,
"embedding_init_range": 0.02,
"eos_token_id": 4,
"hidden_size": 960,
"intermediate_size": 3840,
"mask_token_id": 2,
"max_length": 2048,
"max_protein_length": 50000,
"model_type": "AMPLIFY",
"norm_eps": 1e-05,
"normalized_transformer": false,
"num_attention_heads": 15,
"num_hidden_layers": 32,
"other_special_token_ids": null,
"pad_token_id": 0,
"remove_ambiguous": true,
"torch_dtype": "float32",
"transformers_version": "4.49.0",
"unk_token_id": 1,
"vocab_path": "/home/mila/l/lola.lebreton/AMPLIFY-private/conf/tokenizer/amplify_vocab.txt",
"vocab_size": 32
}