ESMFold2-hf / config.json
Rocketknight1's picture
Rocketknight1 HF Staff
Upload 10 files
bce015e verified
Raw
History Blame Contribute Delete
4.23 kB
{
"architectures": [
"EsmFold2Model"
],
"atom_encoder": {
"head_dim": 32,
"hidden_size": 128,
"intermediate_size": 256,
"model_type": "",
"num_attention_heads": 4,
"num_hidden_layers": 3,
"num_spatial_rope_pairs_per_axis": 2,
"num_uid_rope_pairs": 10,
"output_dim": 384,
"spatial_rope_base_frequency": 20.0,
"uid_rope_base_frequency": 10000.0
},
"atom_feature_dim": 389,
"char_vocab_size": 64,
"chunk_size": 64,
"confidence_head": {
"distogram_bins": 39,
"eps": 1e-06,
"max_dist": 50.75,
"min_dist": 3.25,
"model_type": "",
"num_hidden_layers": 4,
"num_pae_bins": 64,
"num_pde_bins": 64,
"num_plddt_bins": 50
},
"dtype": "float32",
"esmc_config": {
"attention_bias": false,
"attention_dropout": 0.0,
"bos_token_id": 0,
"classifier_dropout": 0.1,
"dtype": "float32",
"eos_token_id": 2,
"head_dim": 64,
"hidden_act": "silu",
"hidden_size": 2560,
"initializer_range": 0.02,
"intermediate_size": 6912,
"mask_token_id": 32,
"max_position_embeddings": 2048,
"mlp_bias": false,
"model_type": "esmc",
"num_attention_heads": 40,
"num_hidden_layers": 80,
"num_key_value_heads": 40,
"pad_token_id": 1,
"rope_parameters": {
"rope_theta": 10000.0,
"rope_type": "default"
},
"tie_word_embeddings": false,
"vocab_size": 64
},
"folding_trunk_num_hidden_layers": 48,
"hidden_size": 384,
"lm_encoder": {
"lm_dropout": 0.25,
"model_type": "",
"num_hidden_layers": 4,
"per_loop_lm_dropout": true
},
"max_atomic_number": 128,
"max_atoms_per_token": 23,
"max_chars": 4,
"model_type": "esmfold2",
"msa_encoder": {
"head_dim": 16,
"hidden_size": 128,
"intermediate_size": 512,
"model_type": "",
"num_attention_heads": 8,
"num_hidden_layers": 4,
"outer_hidden_size": 32,
"outer_product_chunk_size": null
},
"num_loops": 3,
"num_relative_chain_bins": 2,
"num_relative_residx_bins": 32,
"num_res_types": 33,
"pair_transition_intermediate_size": 1024,
"pairwise_hidden_size": 256,
"parcae_num_coda_layers": 2,
"single_inputs_size": 451,
"sliding_window": 128,
"structure_head": {
"diffusion_module": {
"_name_or_path": "",
"architectures": null,
"atom_encoder": {
"_name_or_path": "",
"architectures": null,
"chunk_size_feed_forward": 0,
"dtype": null,
"head_dim": 32,
"hidden_size": 128,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"intermediate_size": 256,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"model_type": "",
"num_attention_heads": 4,
"num_hidden_layers": 3,
"num_spatial_rope_pairs_per_axis": 2,
"num_uid_rope_pairs": 10,
"output_attentions": false,
"output_dim": 768,
"output_hidden_states": false,
"problem_type": null,
"return_dict": true,
"spatial_rope_base_frequency": 20.0,
"uid_rope_base_frequency": 10000.0
},
"chunk_size_feed_forward": 0,
"dtype": null,
"fourier_dim": 256,
"head_dim": 48,
"hidden_size": 768,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"intermediate_size": 1536,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"model_type": "",
"num_attention_heads": 16,
"num_hidden_layers": 12,
"output_attentions": false,
"output_hidden_states": false,
"pair_intermediate_size": 512,
"problem_type": null,
"return_dict": true,
"sigma_data": 16.0
},
"gamma_0": 0.8,
"gamma_min": 1.0,
"inference_exponent": 7.0,
"inference_num_steps": 14,
"inference_sigma_cap": 256.0,
"inference_sigma_max_ratio": 160.0,
"inference_sigma_min_ratio": 0.0004,
"model_type": "",
"noise_scale": 1.003,
"num_diffusion_samples": 32,
"num_distogram_bins": 64,
"step_scale": 1.5
},
"transformers_version": "5.16.0.dev0"
}