Diffusers
Safetensors
English
RFD4ProteinaPipeline
biology
protein-design
biomolecular-design
structure-generation
flow-matching
annotated-cif
Instructions to use nvidia/RFD4-Proteina-Diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use nvidia/RFD4-Proteina-Diffusers with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("nvidia/RFD4-Proteina-Diffusers", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
File size: 1,443 Bytes
a1a523e 5db5138 a1a523e 36e146b a1a523e 36e146b a1a523e 36e146b a1a523e cbd93f5 a1a523e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 | {
"_class_name": "RFD4ProteinaTransformer",
"_diffusers_version": "0.42.0.dev0",
"cath_emb_dim": 32,
"dim_cond": 256,
"feats_cond_seq": [
"time_emb_bb_ca",
"time_emb_local_latents",
"cath_code",
"recycle_remaining"
],
"feats_pair_cond": [
"time_emb_bb_ca",
"time_emb_local_latents"
],
"feats_pair_repr": [
"xt_bb_ca_pair_dists",
"bond_mask",
"bond_order",
"is_aromatic",
"index_group_pair",
"molecule_idx_pair",
"molecule_entity_pair",
"token_bond_path_length",
"canonical_polymer_ordering",
"na_ss_pair"
],
"feats_seq": [
"xt_bb_ca",
"xt_local_latents",
"token_type",
"sequence_probabilities",
"cond_a37coors_nm",
"cond_a37coors_internal_nm",
"rasa_binned",
"secondary_structure",
"hbond_modernized",
"pistack",
"pication",
"saltbridge",
"hydrophobic",
"metal_coord",
"na_ss_state",
"hotspot",
"atom_charge",
"atomic_element",
"coord_group_seq",
"chirality",
"chain_centroid_xyz"
],
"latent_dim": 32,
"nheads": 12,
"nlayers": 14,
"num_sequence_classes": 147,
"num_token_types": 7,
"pair_repr_dim": 384,
"recycle_conditioning_dim": 1,
"seq_sep_dim": 64,
"t_emb_dim": 256,
"token_dim": 768,
"token_type_emb_dim": 32,
"tri_mult_c": 256,
"update_pair_repr_every_n": 2,
"xt_pair_dist_dim": 30,
"xt_pair_dist_max": 3,
"xt_pair_dist_min": 0.1
}
|