Phsntom/ESMFold2-Fast

1{2  "architectures": [3    "ESMFold2Model"4  ],5  "confidence_head": {6    "distogram_bins": 39,7    "enabled": true,8    "folding_trunk": {9      "dropout": 0.25,10      "n_heads": 8,11      "n_layers": 412    },13    "max_dist": 50.75,14    "min_dist": 3.25,15    "num_pae_bins": 64,16    "num_pde_bins": 64,17    "num_plddt_bins": 5018  },19  "d_pair": 256,20  "d_single": 384,21  "disable_msa_features": false,22  "dtype": "float32",23  "esmc_id": "biohub/ESMC-6B",24  "folding_trunk": {25    "dropout": 0.25,26    "n_heads": 8,27    "n_layers": 2428  },29  "force_lm_dropout_during_inference": false,30  "inputs": {31    "atom_encoder": {32      "d_atom": 128,33      "d_token": 768,34      "expansion_ratio": 2,35      "n_blocks": 3,36      "n_heads": 4,37      "n_spatial_rope_pairs_per_axis": 2,38      "n_uid_rope_pairs": 10,39      "spatial_rope_base_frequency": 20.0,40      "swa_window_size": 128,41      "uid_rope_base_frequency": 10000.042    },43    "d_inputs": 45144  },45  "lm_d_model": 2560,46  "lm_dropout": 0.0,47  "lm_encoder": {48    "enabled": true,49    "lm_dropout": 0.25,50    "n_layers": 4,51    "per_loop_lm_dropout": true52  },53  "lm_num_layers": 80,54  "model_type": "esmfold2",55  "msa_encoder": {56    "d_hidden": 32,57    "d_msa": 128,58    "enabled": false,59    "msa_head_width": 32,60    "n_heads_msa": 8,61    "n_layers": 462  },63  "msa_encoder_overwrite": true,64  "n_relative_chain_bins": 2,65  "n_relative_residx_bins": 32,66  "num_diffusion_samples": 32,67  "num_loops": 3,68  "parcae": {69    "coda_n_layers": 2,70    "enabled": true,71    "max_steps": 6,72    "min_steps": 1,73    "poisson_mean": 3.074  },75  "structure_head": {76    "diffusion_module": {77      "atom_num_blocks": 3,78      "atom_num_heads": 4,79      "c_atom": 128,80      "c_s_inputs": 451,81      "c_token": 768,82      "c_z": 256,83      "fourier_dim": 256,84      "relpos_r_max": 32,85      "relpos_s_max": 2,86      "sigma_data": 16.0,87      "token_num_blocks": 12,88      "token_num_heads": 16,89      "transition_multiplier": 290    },91    "distogram_bins": 64,92    "gamma_0": 0.8,93    "gamma_min": 1.0,94    "inference_num_steps": 14,95    "inference_p": 7.0,96    "inference_s_max": 160.0,97    "inference_s_min": 0.0004,98    "noise_scale": 1.003,99    "step_scale": 1.5,100    "train_noise_log_mean": -1.2,101    "train_noise_log_std": 1.5102  },103  "transformers_version": "4.57.6",104  "type": "release"105}