| { |
| "architectures": [ |
| "ESMFold2ExperimentalModel" |
| ], |
| "auto_map": { |
| "AutoConfig": "modeling_fastplms.ESMFold2Config", |
| "AutoModel": "modeling_fastplms.ESMFold2ExperimentalModel", |
| "AutoModelForSequenceClassification": "modeling_fastplms.ESMFold2ExperimentalForSequenceClassification", |
| "AutoModelForTokenClassification": "modeling_fastplms.ESMFold2ExperimentalForTokenClassification" |
| }, |
| "confidence_head": { |
| "distogram_bins": 128, |
| "enabled": true, |
| "folding_trunk": { |
| "dropout": 0.25, |
| "n_heads": 8, |
| "n_layers": 4 |
| }, |
| "max_dist": 52.0, |
| "min_dist": 2.0, |
| "num_pae_bins": 64, |
| "num_pde_bins": 64, |
| "num_plddt_bins": 50 |
| }, |
| "d_pair": 256, |
| "d_single": 384, |
| "disable_msa_features": true, |
| "dtype": "float32", |
| "esmc_id": "Synthyra/ESMplusplus_large", |
| "fastplms_checkpoint_hash": "4e3baff4574120dfa58a25f34c910a05943e343ce63ad268775d2ce33b3b8c64", |
| "fastplms_checkpoint_repo_id": "Synthyra/ESMFold2-600", |
| "fastplms_model_id": "esmfold2_600", |
| "fastplms_runtime_bundle_sha256": "2ef7cccec8dfd129c6621d7bbb4b9f91a0ed1be1fd3f62d670259a4a9e199af0", |
| "folding_trunk": { |
| "dropout": 0.25, |
| "n_heads": 8, |
| "n_layers": 24 |
| }, |
| "force_lm_dropout_during_inference": false, |
| "inputs": { |
| "atom_encoder": { |
| "d_atom": 128, |
| "d_token": 768, |
| "expansion_ratio": 2, |
| "n_blocks": 3, |
| "n_heads": 4, |
| "n_spatial_rope_pairs_per_axis": 2, |
| "n_uid_rope_pairs": 10, |
| "spatial_rope_base_frequency": 20.0, |
| "swa_window_size": 128, |
| "uid_rope_base_frequency": 10000.0 |
| }, |
| "d_inputs": 451 |
| }, |
| "lm_d_model": 1152, |
| "lm_dropout": 0.0, |
| "lm_num_layers": 36, |
| "model_type": "esmfold2", |
| "msa_conditioning": false, |
| "msa_encoder": { |
| "d_hidden": 32, |
| "d_msa": 128, |
| "enabled": false, |
| "msa_head_width": 32, |
| "n_heads_msa": 8, |
| "n_layers": 4 |
| }, |
| "n_relative_chain_bins": 2, |
| "n_relative_residx_bins": 32, |
| "num_diffusion_samples": 16, |
| "num_loops": 3, |
| "structure_head": { |
| "diffusion_module": { |
| "atom_num_blocks": 3, |
| "atom_num_heads": 4, |
| "c_atom": 128, |
| "c_s_inputs": 451, |
| "c_token": 768, |
| "c_z": 256, |
| "fourier_dim": 256, |
| "relpos_r_max": 32, |
| "relpos_s_max": 2, |
| "sigma_data": 16.0, |
| "token_num_blocks": 12, |
| "token_num_heads": 16, |
| "transition_multiplier": 2 |
| }, |
| "distogram_bins": 128, |
| "gamma_0": 0.605, |
| "gamma_min": 1.107, |
| "inference_num_steps": 15, |
| "inference_p": 8.0, |
| "inference_s_max": 160.0, |
| "inference_s_min": 0.0004, |
| "noise_scale": 0.901, |
| "step_scale": 1.638, |
| "train_noise_log_mean": -1.2, |
| "train_noise_log_std": 1.5 |
| }, |
| "transformers_version": "4.57.6", |
| "type": "experimental", |
| "confidence_head_release": { |
| "name": "v1", |
| "head_sha256": "e84726a050722e1b722712c87d17a5388bd3699e2520e4a59abb1d828dfb8de7", |
| "base_weight_sha256": "11a53c1b4700b6c62a5a464fc3ec7076160c19e8584157f88e13275116cfd602", |
| "training_url": "https://wandb.ai/lhallee/fastplms-confidence/runs/820d2cfa56c0", |
| "updates": 780, |
| "evaluation_request_sha256": "1874e73d7e3acfc0766ff38b51d4eb09e735610c41cbe388b6b19388fa8c10db" |
| } |
| } |
|
|
