kierandidi commited on
Commit
7694a9e
·
verified ·
1 Parent(s): 428e1dc

Model card: the autoencoder component is registered as vae

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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` + `autoencoder` + `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,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/`, `autoencoder/` | 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 |
 
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 |