| { |
| "architecture": "pxdesign", |
| "atom_feature_size": 385, |
| "condition_embeddings": { |
| "add_feature_size": 4, |
| "atom_hidden_size": 128, |
| "atom_pair_hidden_size": 16, |
| "design_feature_size": 1, |
| "hotspot_size": 1, |
| "num_atom_encoder_attention_heads": 4, |
| "num_atom_encoder_blocks": 3, |
| "num_key_atoms": 128, |
| "num_query_atoms": 32, |
| "plddt_size": 1, |
| "token_hidden_size": 384, |
| "vocab_size": 72 |
| }, |
| "denoiser": { |
| "atom_hidden_size": 128, |
| "atom_pair_hidden_size": 16, |
| "num_atom_decoder_attention_heads": 4, |
| "num_atom_decoder_blocks": 4, |
| "num_atom_encoder_attention_heads": 4, |
| "num_atom_encoder_blocks": 4, |
| "num_key_atoms": 128, |
| "num_query_atoms": 32, |
| "num_token_encoder_attention_heads": 16, |
| "num_token_encoder_blocks": 16, |
| "sigma_data": 16.0, |
| "token_hidden_size": 768 |
| }, |
| "num_template_bins": 64, |
| "pair_size": 128, |
| "sampler": { |
| "churn_rate": 1.0, |
| "max_noise_level": 160.0, |
| "max_step_scale": 2.5, |
| "min_churn_noise_level": 0.01, |
| "min_noise_level": 0.0004, |
| "min_step_scale": 1.0, |
| "noise_scale": 1.003, |
| "rho": 7.0, |
| "step_scale_schedule": "piecewise_65" |
| }, |
| "single_size": 384, |
| "token_feature_size": 449 |
| } |
|
|
