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
Model card: the autoencoder component is registered as vae
Browse files
README.md
CHANGED
|
@@ -47,7 +47,7 @@ This repository is the [Diffusers](https://github.com/huggingface/diffusers) rel
|
|
| 47 |
| | Details |
|
| 48 |
|---|---|
|
| 49 |
| **Model** | Joint sequence–structure generation with a matched flow transformer and autoencoder |
|
| 50 |
-
| **Pipeline** | `RFD4ProteinaPipeline`: `transformer` + `
|
| 51 |
| **Inputs** | Annotated CIF structures, prepared with [`rfd4-proteina-prep`](https://huggingface.co/nvidia/RFD4-Proteina-prep) |
|
| 52 |
| **Outputs** | Atom37 coordinates and residue identities, written back to annotated CIF |
|
| 53 |
| **Weights** | EMA weights, 434.3 million parameters, stored in FP32 (1.74 GB); runs in BF16 |
|
|
@@ -139,7 +139,7 @@ The same seed can end in a different design in BF16 and FP32, because the sample
|
|
| 139 |
| Path | Use |
|
| 140 |
|---|---|
|
| 141 |
| `model_index.json` | Pipeline definition for `RFD4ProteinaPipeline.from_pretrained` |
|
| 142 |
-
| `transformer/`, `
|
| 143 |
| `scheduler/` | `RFD4ProductSpaceScheduler` config with the released sampler recipe and recycle schedule |
|
| 144 |
| [`provenance.json`](provenance.json) | Source bundle hashes, sampler recipe and converter version |
|
| 145 |
| `examples/` | Studio inputs, generated designs and previews |
|
|
|
|
| 47 |
| | Details |
|
| 48 |
|---|---|
|
| 49 |
| **Model** | Joint sequence–structure generation with a matched flow transformer and autoencoder |
|
| 50 |
+
| **Pipeline** | `RFD4ProteinaPipeline`: `transformer` + `vae` + `scheduler` |
|
| 51 |
| **Inputs** | Annotated CIF structures, prepared with [`rfd4-proteina-prep`](https://huggingface.co/nvidia/RFD4-Proteina-prep) |
|
| 52 |
| **Outputs** | Atom37 coordinates and residue identities, written back to annotated CIF |
|
| 53 |
| **Weights** | EMA weights, 434.3 million parameters, stored in FP32 (1.74 GB); runs in BF16 |
|
|
|
|
| 139 |
| Path | Use |
|
| 140 |
|---|---|
|
| 141 |
| `model_index.json` | Pipeline definition for `RFD4ProteinaPipeline.from_pretrained` |
|
| 142 |
+
| `transformer/`, `vae/` | Denoiser and autoencoder configs and safetensors weights |
|
| 143 |
| `scheduler/` | `RFD4ProductSpaceScheduler` config with the released sampler recipe and recycle schedule |
|
| 144 |
| [`provenance.json`](provenance.json) | Source bundle hashes, sampler recipe and converter version |
|
| 145 |
| `examples/` | Studio inputs, generated designs and previews |
|