| { |
| "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_small", |
| "fastplms_checkpoint_hash": "bb2eaf6389e1d58f5117ca3173a5ee73d73548cda5c89f95f85c4a2280ad0e7e", |
| "fastplms_checkpoint_repo_id": "Synthyra/ESMFold2-300", |
| "fastplms_model_id": "esmfold2_300", |
| "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": 960, |
| "lm_dropout": 0.0, |
| "lm_num_layers": 30, |
| "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": "40fd7f3d82fcefe8ad20ab2b32a37a68a84b54a527a4bce6eb9437bad2b77e31", |
| "base_weight_sha256": "44d6797c5efebf24753d502b40950e0874871c96ceea14f2d7f7e39cebac67fd", |
| "training_url": "https://wandb.ai/lhallee/fastplms-confidence/runs/9558b6d23daf", |
| "updates": 780, |
| "evaluation_request_sha256": "be5fb9a4b65a0c2abfa040069ffd4344a9d220ef8e595711c4849a62401e281a" |
| } |
| } |
|
|
