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
Drop transformer options that are constant for the release
Browse files- transformer/config.json +0 -5
transformer/config.json
CHANGED
|
@@ -51,21 +51,16 @@
|
|
| 51 |
"latent_dim": 32,
|
| 52 |
"nheads": 12,
|
| 53 |
"nlayers": 14,
|
| 54 |
-
"noise_align_chirality": true,
|
| 55 |
"num_sequence_classes": 147,
|
| 56 |
"num_token_types": 7,
|
| 57 |
"pair_repr_dim": 384,
|
| 58 |
-
"recycle": true,
|
| 59 |
"recycle_conditioning_dim": 1,
|
| 60 |
"seq_sep_dim": 64,
|
| 61 |
"t_emb_dim": 256,
|
| 62 |
"token_dim": 768,
|
| 63 |
"token_type_emb_dim": 32,
|
| 64 |
-
"tri_mult_bias": false,
|
| 65 |
"tri_mult_c": 256,
|
| 66 |
-
"update_pair_repr": true,
|
| 67 |
"update_pair_repr_every_n": 2,
|
| 68 |
-
"use_tri_mult": true,
|
| 69 |
"xt_pair_dist_dim": 30,
|
| 70 |
"xt_pair_dist_max": 3,
|
| 71 |
"xt_pair_dist_min": 0.1
|
|
|
|
| 51 |
"latent_dim": 32,
|
| 52 |
"nheads": 12,
|
| 53 |
"nlayers": 14,
|
|
|
|
| 54 |
"num_sequence_classes": 147,
|
| 55 |
"num_token_types": 7,
|
| 56 |
"pair_repr_dim": 384,
|
|
|
|
| 57 |
"recycle_conditioning_dim": 1,
|
| 58 |
"seq_sep_dim": 64,
|
| 59 |
"t_emb_dim": 256,
|
| 60 |
"token_dim": 768,
|
| 61 |
"token_type_emb_dim": 32,
|
|
|
|
| 62 |
"tri_mult_c": 256,
|
|
|
|
| 63 |
"update_pair_repr_every_n": 2,
|
|
|
|
| 64 |
"xt_pair_dist_dim": 30,
|
| 65 |
"xt_pair_dist_max": 3,
|
| 66 |
"xt_pair_dist_min": 0.1
|