Upload 10 files

config.json

@@ -3,15 +3,14 @@
     "EsmFold2Model"
   ],
   "atom_encoder": {
-    "expansion_ratio": 2,
-    "ffn_intermediate_size": 256,
     "head_dim": 32,
     "hidden_size": 128,
+    "intermediate_size": 256,
     "model_type": "",
-    "n_spatial_rope_pairs_per_axis": 2,
-    "n_uid_rope_pairs": 10,
     "num_attention_heads": 4,
     "num_hidden_layers": 3,
+    "num_spatial_rope_pairs_per_axis": 2,
+    "num_uid_rope_pairs": 10,
     "output_dim": 384,
     "spatial_rope_base_frequency": 20.0,
     "uid_rope_base_frequency": 10000.0
@@ -74,20 +73,19 @@
   "model_type": "esmfold2",
   "msa_encoder": {
     "divide_outer_before_proj": false,
-    "head_width": 16,
+    "head_dim": 16,
     "hidden_size": 128,
+    "intermediate_size": 512,
     "model_type": "",
     "num_attention_heads": 8,
     "num_hidden_layers": 4,
     "outer_hidden_size": 32,
     "outer_product_chunk_size": null,
-    "overwrite": true,
-    "transition_intermediate_size": 512
+    "overwrite": true
   },
-  "n_relative_chain_bins": 2,
-  "n_relative_residx_bins": 32,
-  "num_diffusion_samples": 32,
   "num_loops": 3,
+  "num_relative_chain_bins": 2,
+  "num_relative_residx_bins": 32,
   "num_res_types": 33,
   "pair_transition_intermediate_size": 1024,
   "pairwise_hidden_size": 256,
@@ -103,24 +101,23 @@
         "architectures": null,
         "chunk_size_feed_forward": 0,
         "dtype": null,
-        "expansion_ratio": 2,
-        "ffn_intermediate_size": 256,
         "head_dim": 32,
         "hidden_size": 128,
         "id2label": {
           "0": "LABEL_0",
           "1": "LABEL_1"
         },
+        "intermediate_size": 256,
         "is_encoder_decoder": false,
         "label2id": {
           "LABEL_0": 0,
           "LABEL_1": 1
         },
         "model_type": "",
-        "n_spatial_rope_pairs_per_axis": 2,
-        "n_uid_rope_pairs": 10,
         "num_attention_heads": 4,
         "num_hidden_layers": 3,
+        "num_spatial_rope_pairs_per_axis": 2,
+        "num_uid_rope_pairs": 10,
         "output_attentions": false,
         "output_dim": 768,
         "output_hidden_states": false,
@@ -132,39 +129,40 @@
       "chunk_size_feed_forward": 0,
       "dtype": null,
       "fourier_dim": 256,
+      "head_dim": 48,
+      "hidden_size": 768,
       "id2label": {
         "0": "LABEL_0",
         "1": "LABEL_1"
       },
+      "intermediate_size": 1536,
       "is_encoder_decoder": false,
       "label2id": {
         "LABEL_0": 0,
         "LABEL_1": 1
       },
       "model_type": "",
+      "num_attention_heads": 16,
+      "num_hidden_layers": 12,
       "output_attentions": false,
       "output_hidden_states": false,
+      "pair_intermediate_size": 512,
       "problem_type": null,
       "return_dict": true,
-      "sigma_data": 16.0,
-      "token_hidden_size": 768,
-      "token_num_blocks": 12,
-      "token_num_heads": 16,
-      "token_transition_intermediate_size": 1536,
-      "transition_multiplier": 2
+      "sigma_data": 16.0
     },
-    "distogram_bins": 64,
     "gamma_0": 0.8,
     "gamma_min": 1.0,
+    "inference_exponent": 7.0,
     "inference_num_steps": 14,
-    "inference_p": 7.0,
-    "inference_s_max": 160.0,
-    "inference_s_min": 0.0004,
-    "max_inference_sigma": 256.0,
+    "inference_sigma_cap": 256.0,
+    "inference_sigma_max_ratio": 160.0,
+    "inference_sigma_min_ratio": 0.0004,
     "model_type": "",
     "noise_scale": 1.003,
+    "num_diffusion_samples": 32,
+    "num_distogram_bins": 64,
     "step_scale": 1.5
   },
-  "transformers_version": "5.14.0.dev0",
-  "transition_expansion_ratio": 4
+  "transformers_version": "5.15.0.dev0"
 }

model-00001-of-00006.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cd7389eeb251f3afd6d16d483e1b9651f973e89ca526335d7308c3f7ad5c6418
-size 4955677724
+oid sha256:fdd94e62ad71ed144ebdc5561d07dd8ace66aec0b195928792e15ef784ce8cbd
+size 4955672068