Upload converted Boltz-2 checkpoint

README.md

@@ -0,0 +1,42 @@
+---
+library_name: "ofoldx"
+tags:
+  - "biology"
+  - "biomolecular-design"
+  - "protein"
+  - "rna"
+  - "dna"
+  - "model"
+  - "boltz2"
+  - "structure-prediction"
+  - "protein-structure"
+artifact_kind: "model"
+repo_id: "oteam/boltz2"
+pipeline_tag: "other"
+task: "structure_prediction"
+widget:
+  - pipeline_tag: "other"
+    task: "structure_prediction"
+    example_title: "Single-chain protein"
+    text: ">A\nMKTAYIAKQRQISFVKSHFSRQDILD"
+    input_format: "fasta"
+  - pipeline_tag: "other"
+    task: "structure_prediction"
+    example_title: "Protein complex"
+    text: ">A\nMKTAYIAKQRQISFVKSHFSRQDILD\n>B\nGSHMRYFVTAVSRPGRGEPRFI"
+    input_format: "fasta"
+---
+
+# boltz2
+
+This repository contains a biomolecular `model` artifact.
+It serves the `structure_prediction` task.
+The architecture key is `boltz2`.
+
+The artifact is described by `manifest.json`; runtime configuration and weights stay in their own files.
+
+```python
+from ofoldx import AutoModel
+
+model = AutoModel.from_pretrained("oteam/boltz2")
+```

config.json

@@ -0,0 +1,109 @@
+{
+  "single_size": 384,
+  "pair_size": 128,
+  "atom_hidden_size": 128,
+  "atom_pair_hidden_size": 16,
+  "atom_feature_size": 388,
+  "num_bins": 64,
+  "num_distograms": 1,
+  "fourier_embedding_size": 256,
+  "relative_position_size": 139,
+  "num_bond_types": 7,
+  "initializer_range": 0.02,
+  "recycling_steps": 3,
+  "recycling_detach": true,
+  "fix_sym_check": true,
+  "cyclic_pos_enc": true,
+  "use_templates": true,
+  "use_templates_v2": true,
+  "bond_type_feature": true,
+  "token_level_confidence": true,
+  "predict_bfactor": true,
+  "add_s_to_z_prod": true,
+  "add_s_input_to_s": true,
+  "add_z_input_to_z": true,
+  "aggregate_distogram": true,
+  "conditioning_cutoff_min": 4.0,
+  "conditioning_cutoff_max": 20.0,
+  "min_distance": 2.0,
+  "max_distance": 22.0,
+  "input_embedder": {
+    "num_atom_encoder_blocks": 3,
+    "num_atom_encoder_attention_heads": 4,
+    "vocab_size": 33,
+    "msa_profile_size": 34,
+    "num_method_types": 12,
+    "num_molecule_types": 4,
+    "add_method_conditioning": true,
+    "add_modified_flag": true,
+    "add_cyclic_flag": true,
+    "add_mol_type_feat": true
+  },
+  "trunk": {
+    "num_pairformer_blocks": 64,
+    "num_pair_attention_heads": 16,
+    "num_msa_blocks": 4,
+    "msa_size": 64,
+    "msa_feature_size": 36,
+    "msa_head_dim": 32,
+    "num_msa_attention_heads": 8,
+    "outer_product_hidden_size": 32,
+    "triangle_attention_hidden_size": 32,
+    "num_triangle_attention_heads": 4,
+    "use_paired_feature": true,
+    "operator_backend": "auto",
+    "dropout": 0.25,
+    "msa_dropout": 0.15
+  },
+  "template": {
+    "hidden_size": 64,
+    "num_blocks": 2,
+    "num_distogram_bins": 38,
+    "vocab_size": 33,
+    "expansion": 4,
+    "triangle_attention_hidden_size": 32,
+    "num_triangle_attention_heads": 4,
+    "min_distance": 3.25,
+    "max_distance": 50.75
+  },
+  "diffusion": {
+    "num_token_transformer_blocks": 24,
+    "num_token_transformer_heads": 16,
+    "num_atom_encoder_blocks": 3,
+    "num_atom_decoder_blocks": 3,
+    "num_atom_attention_heads": 4,
+    "atom_hidden_size": 128,
+    "atom_pair_hidden_size": 16,
+    "num_query_atoms": 32,
+    "num_key_atoms": 128,
+    "num_conditioning_transitions": 2,
+    "sigma_data": 16.0,
+    "alignment_reverse_diff": true,
+    "operator_backend": "auto"
+  },
+  "sampler": {
+    "solver": "euler",
+    "num_sampling_steps": 200,
+    "rho": 7.0,
+    "max_noise_level": 160.0,
+    "min_noise_level": 0.0004,
+    "step_scale": 1.5,
+    "noise_scale": 1.003,
+    "churn_rate": 0.8
+  },
+  "confidence": {
+    "num_pairformer_blocks": 8,
+    "num_pair_attention_heads": 16,
+    "num_distogram_bins": 64,
+    "max_distance": 22.0,
+    "num_plddt_bins": 50,
+    "num_pae_bins": 64,
+    "num_pde_bins": 64,
+    "num_bfactor_bins": 64,
+    "add_s_to_z_prod": true,
+    "add_s_input_to_s": true,
+    "add_z_input_to_z": true,
+    "use_separate_heads": true,
+    "token_level": true
+  }
+}

manifest.json

@@ -0,0 +1,18 @@
+{
+  "format": "biomolecular_pretrained",
+  "format_version": 1,
+  "kind": "model",
+  "producer": {
+    "package": "ofoldx",
+    "version": "0.1.dev1+g9a78c7e44.d20260609"
+  },
+  "entrypoint": "ofoldx.models.boltz2.model.Boltz2Model",
+  "architecture": "boltz2",
+  "task": "structure_prediction",
+  "files": {
+    "config": "config.json"
+  },
+  "weights": {
+    "default": "model.safetensors"
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b9873bc3a5e19c7a1a36683682a68619440d29b8cf9a78eab807ffb1602b91ac
+size 2027725444