Add VenusFold EMA weights

README.md

@@ -0,0 +1,23 @@
+---
+license: apache-2.0
+library_name: pytorch
+tags:
+  - protein-folding
+  - protein-structure
+  - dna
+  - rna
+  - ligand
+---
+
+# VenusFold
+
+Inference weights for VenusFold biomolecular structure prediction.
+
+- Weight variant: EMA (0.999)
+- Parameters: 464,442,431
+- Activation: SiLU
+- Default inference: 200 diffusion steps and 10 recycling cycles
+
+Code and usage: [ai4protein/VenusFold](https://github.com/ai4protein/VenusFold)
+
+VenusFold is a research model and is not validated for clinical or diagnostic use.

config.yaml

@@ -0,0 +1,18 @@
+model_name: venusfold
+load_strict: true
+dtype: bf16
+triangle_attention: cuequivariance
+triangle_multiplicative: cuequivariance
+enable_tf32: true
+enable_efficient_fusion: true
+enable_diffusion_shared_vars_cache: false
+use_msa: true
+use_template: false
+use_rna_msa: true
+num_workers: 0
+model:
+  N_cycle: 10
+  local_activation: silu
+sample_diffusion:
+  N_sample: 5
+  N_step: 200

model.json

@@ -0,0 +1,8 @@
+{
+  "file": "model.safetensors",
+  "sha256": "01e1c1fd5bf5d1d1044b978da8ae4f05ef9fa844569a7fb8f84f89297a6fc567",
+  "parameter_count": 464442431,
+  "dtype": [
+    "torch.float32"
+  ]
+}

model.safetensors

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