Hugging Face's logo

Diffusers documentation

RFD4ProteinaAutoencoder

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.42.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

RFD4ProteinaAutoencoder

Autoencoder dari RFD4-Proteína. encode memetakan koordinat atom37 ke latens lokal per-token, dan decode mengubah latens dan posisi backbone kembali menjadi struktur atom37 dan identitas residu. Ini didaftarkan dalam pipeline sebagai vae.

from diffusers import RFD4ProteinaAutoencoder

vae = RFD4ProteinaAutoencoder.from_pretrained("nvidia/RFD4-Proteina-Diffusers", subfolder="vae")

RFD4ProteinaAutoencoder

kelas diffusers.RFD4ProteinaAutoencoder

< >

( atom37_mask: Sequenceallowed_residue_types: Sequencelatent_dim: int = 32n_residue_types: int = 147num_token_types: int = 7nlayers: int = 12token_dim: int = 768nheads: int = 12dim_cond: int = 128pair_repr_dim: int = 256token_type_emb_dim: int = 32encoder_feats_seq: tuple = ('token_type', 'x1_a37coors_nm', 'x1_a37coors_nm_rel', 'chain_idx_seq', 'incident_bond_type_counts', 'atom37_element_charge_embedding')decoder_feats_seq: tuple = ('ca_coors_nm', 'z_latent_seq', 'token_type', 'cond_residue_type') )

Parameter

  • latent_dim (int, defaultnya 32) — Lebar laten lokal per token.
  • n_residue_types (int, defaultnya 147) — Ukuran kosakata residu.
  • num_token_types (int, defaultnya 7) — Ukuran kosakata tipe token: protein, RNA, DNA, atomized, unknown, glikan, dan asam amino non-kanonik.
  • allowed_residue_types (list[list[bool]]) — tabel (num_token_types, n_residue_types) dari kelas residu yang dapat didekodekan oleh setiap tipe token.
  • nlayers (int, defaultnya 12) — Lapisan trunk di masing-masing encoder dan decoder.
  • token_dim (int, defaultnya 768) — Lebar representasi token.
  • nheads (int, defaultnya 12) — Jumlah kepala perhatian per lapisan.
  • dim_cond (int, defaultnya 128) — Lebar dari vektor kondisioning (tidak digunakan, nol).
  • pair_repr_dim (int, defaultnya 256) — Lebar pasangan yang digunakan untuk membangun lapisan trunk. Autoencoder yang dirilis tidak menggunakan bias pasangan, sehingga parameter ini tidak menyimpan parameter apa pun.
  • token_type_emb_dim (int, defaultnya 32) — Lebar embedding tipe token.
  • encoder_feats_seq, decoder_feats_seq (list[str]) — Fitur yang diproyeksikan ke input encoder dan decoder.
  • atom37_mask (list[list[bool]]) — Slot atom37 yang valid untuk setiap tipe residu, dengan bentuk (n_residue_types, 37).

Encoder molekul berpasangan dan decoder atom37.

Koordinat yang diberikan ke dan dikembalikan oleh model ini dalam satuan nanometer.

encode

< >

( coords_nm: Tensorpadding_mask: Tensortoken_type: typing.Optional[torch.Tensor] = Noneatom37_mask: typing.Optional[torch.Tensor] = Noneconditioning: dict[str, typing.Any] | None = Nonegenerator: typing.Union[torch.Generator, list[torch.Generator], NoneType] = Nonereturn_dict: bool = True ) → RFD4ProteinaEncoderOutput or tuple

Parameter

  • coords_nm (torch.Tensor) — koordinat atom37 dalam nanometer, bentuk (batch, tokens, 37, 3).
  • padding_mask (torch.Tensor) — (batch, tokens), bernilai true di mana token tersebut asli, bukan padding.
  • token_type (torch.Tensor, opsional) — kode token-type (batch, tokens).
  • atom37_mask (torch.Tensor, opsional) — (batch, tokens, 37), bernilai true di mana slot atom digunakan.
  • conditioning (dict[str, Any], opsional) — Tensor topologi dan conditioning yang tersisa, di bawah kunci yang dihasilkan oleh preprocessor.
  • generator (torch.Generator atau list[torch.Generator], opsional) — Satu generator per baris batch untuk sampel laten stokastik yang independen. Tanpa generator, pengambilan sampel dilakukan dari aliran global.
  • return_dict (bool, defaultnya True) — Mengembalikan output bertipe, atau tuple dari field-fieldnya.

Mengembalikan

RFD4ProteinaEncoderOutput atau tuple

Tensor laten, rata-rata, dan skala log dengan bentuk (batch, tokens, latent_dim).

Enkode koordinat yang telah disiapkan dan kondisioning menjadi latens lokal per-token yang disampel.

decode

< >

( z_latent: Tensorca_coors_nm: Tensorpadding_mask: Tensortoken_type: typing.Optional[torch.Tensor] = Noneatom37_mask: typing.Optional[torch.Tensor] = Noneconditioning: dict[str, typing.Any] | None = Nonereturn_dict: bool = True ) → RFD4ProteinaDecoderOutput or tuple

Parameter

  • z_latent (torch.Tensor) — latens lokal per token (batch, tokens, latent_dim).
  • ca_coors_nm (torch.Tensor) — posisi CA backbone dalam nanometer, bentuk (batch, tokens, 3).
  • padding_mask (torch.Tensor) — (batch, tokens), bernilai true di mana token tersebut asli, bukan padding.
  • token_type (torch.Tensor, opsional) — kode token-type (batch, tokens).
  • atom37_mask (torch.Tensor, opsional) — (batch, tokens, 37), bernilai true di mana slot atom digunakan.
  • conditioning (dict[str, Any], opsional) — Tensor topologi dan kondisioning lainnya, di bawah kunci yang dihasilkan oleh preprocessor. Kondisi urutan dibaca dari sini.
  • return_dict (bool, defaultnya True) — Mengembalikan output bertipe, atau tuple dari field-fieldnya.

Mengembalikan

RFD4ProteinaDecoderOutput atau tuple

Koordinat (batch, tokens, 37, 3) dalam nanometer, tipe residu (batch, tokens), emission mask (batch, tokens) dan atom mask (batch, tokens, 37). Koordinat atom yang tidak diemisi bernilai nol.

Decode latens per-token dan posisi backbone menjadi struktur atom37.

forward

< >

( z_latent: Tensorca_coors_nm: Tensorpadding_mask: Tensortoken_type: typing.Optional[torch.Tensor] = Noneatom37_mask: typing.Optional[torch.Tensor] = Noneconditioning: dict[str, typing.Any] | None = Nonereturn_dict: bool = True ) → RFD4ProteinaDecoderOutput or tuple

Parameter

  • z_latent (torch.Tensor) — latens lokal per token (batch, tokens, latent_dim).
  • ca_coors_nm (torch.Tensor) — posisi CA backbone dalam nanometer, bentuk (batch, tokens, 3).
  • padding_mask (torch.Tensor) — (batch, tokens), bernilai true di mana token tersebut asli, bukan padding.
  • token_type (torch.Tensor, opsional) — kode token-type (batch, tokens).
  • atom37_mask (torch.Tensor, opsional) — (batch, tokens, 37), bernilai true di mana slot atom digunakan.
  • conditioning (dict[str, Any], opsional) — Tensor topologi dan kondisioning lainnya.
  • return_dict (bool, defaultnya True) — Mengembalikan output bertipe, atau tuple dari field-fieldnya.

Mengembalikan

RFD4ProteinaDecoderOutput atau tuple

Koordinat yang didekodekan, jenis residu, dan mask atom/emit.

Decode latens per-token dan posisi backbone; setara dengan memanggil decode().

RFD4ProteinaEncoderOutput

kelas diffusers.models.autoencoders.autoencoder_rfd4_proteina.RFD4ProteinaEncoderOutput

< >

( latent: Tensormean: typing.Optional[torch.Tensor] = Nonelog_scale: typing.Optional[torch.Tensor] = None )

Parameter

  • latent (torch.Tensor) — Sampel mean + noise * exp(log_scale), nol di luar token latent yang dipancarkan.
  • mean (torch.Tensor) — rata-rata Gaussian, nol di luar token laten yang dipancarkan.
  • log_scale (torch.Tensor) — Deviasi standar log, nol di luar token laten yang dihasilkan.

Parameter laten yang disampelkan dan Gaussian, masing-masing berbentuk (batch, tokens, latent_dim).

RFD4ProteinaDecoderOutput

kelas diffusers.models.autoencoders.autoencoder_rfd4_proteina.RFD4ProteinaDecoderOutput

< >

( coordinates: Tensorresidue_type: typing.Optional[torch.Tensor] = Noneemit_mask: typing.Optional[torch.Tensor] = Noneatom_mask: typing.Optional[torch.Tensor] = None )

Parameter

  • coordinates (torch.Tensor) — Koordinat (batch, tokens, 37, 3) dalam nanometer; atom yang tidak dipancarkan bernilai nol.
  • residue_type (torch.Tensor) — Indeks kosakata residu dengan bentuk (batch, tokens).
  • emit_mask (torch.Tensor) — Mask token yang dipancarkan berbentuk (batch, tokens).
  • atom_mask (torch.Tensor) — Mask atom valid boolean dengan bentuk (batch, tokens, 37).

Koordinat molekul yang didekode dan topologi atom37 diskrit.

Update on GitHub

←QwenImageTransformer2DModel RFD4ProteinaTransformer→