Diffusers documentation
DiTTransformer2DModel
DiTTransformer2DModel
Model Transformer untuk data seperti gambar dari DiT.
DiTTransformer2DModel
kelas diffusers.DiTTransformer2DModel
< source >( num_attention_heads: int = 16attention_head_dim: int = 72in_channels: int = 4out_channels: int | None = Nonenum_layers: int = 28dropout: float = 0.0norm_num_groups: int = 32attention_bias: bool = Truesample_size: int = 32patch_size: int = 2activation_fn: str = 'gelu-approximate'num_embeds_ada_norm: int | None = 1000upcast_attention: bool = Falsenorm_type: str = 'ada_norm_zero'norm_elementwise_affine: bool = Falsenorm_eps: float = 1e-05 )
Parameter
- num_attention_heads (int, opsional, default 16) — Jumlah head yang digunakan untuk multi-head attention.
- attention_head_dim (int, opsional, default 72) — Jumlah channel di setiap head.
- in_channels (int, defaultnya 4) — Jumlah channel pada input.
- out_channels (int, opsional) — Jumlah channel pada output. Tentukan parameter ini jika jumlah channel output berbeda dari input.
- num_layers (int, opsional, default 28) — Jumlah lapisan blok Transformer yang digunakan.
- dropout (float, opsional, default 0.0) — Probabilitas dropout yang digunakan di dalam blok Transformer.
- norm_num_groups (int, opsional, default 32) — Jumlah grup untuk normalisasi grup di dalam blok Transformer.
- attention_bias (bool, opsional, default True) — Mengonfigurasi apakah perhatian pada blok Transformer harus memuat parameter bias.
- sample_size (int, defaultnya 32) — Lebar dari gambar laten. Parameter ini tetap selama pelatihan.
- patch_size (int, defaultnya 2) — Ukuran patch yang diproses oleh model, relevan untuk arsitektur yang bekerja pada data non-berurutan.
- activation_fn (str, opsional, default “gelu-approximate”) — Fungsi aktivasi yang digunakan dalam jaringan feed-forward di dalam blok Transformer.
- num_embeds_ada_norm (int, opsional, default 1000) — Jumlah embedding untuk AdaLayerNorm, yang tetap selama pelatihan dan memengaruhi jumlah langkah denoising maksimum selama inferensi.
- upcast_attention (bool, opsional, default False) — Jika true, menaikkan dimensi mekanisme perhatian untuk potensi peningkatan kinerja.
- norm_type (str, opsional, default “ada_norm_zero”) — Menentukan jenis normalisasi yang digunakan, dapat berupa ‘ada_norm_zero’.
- norm_elementwise_affine (bool, opsional, default False) — Jika true, mengaktifkan parameter afine per elemen di lapisan normalisasi.
- norm_eps (float, opsional, default 1e-5) — Konstanta kecil yang ditambahkan ke penyebut pada lapisan normalisasi untuk mencegah pembagian dengan nol.
Model Transformer 2D yang diperkenalkan dalam DiT (https://huggingface.co/papers/2212.09748).
forward
< source >( hidden_states: Tensortimestep: typing.Optional[torch.LongTensor] = Noneclass_labels: typing.Optional[torch.LongTensor] = Nonecross_attention_kwargs: dict = Nonereturn_dict: bool = True )
Parameter
- hidden_states (
torch.LongTensordengan bentuk(batch size, num latent pixels)jika diskrit,torch.FloatTensordengan bentuk(batch size, channel, height, width)jika kontinu) — Inputhidden_states. - timestep (
torch.LongTensor, opsional) — Digunakan untuk menunjukkan langkah denoising. Timestep opsional yang diterapkan sebagai embedding diAdaLayerNorm. - class_labels (
torch.LongTensordengan bentuk(batch size, num classes), opsional) — Digunakan untuk menunjukkan kondisioning label kelas. Label kelas opsional yang akan diterapkan sebagai embedding diAdaLayerZeroNorm. - cross_attention_kwargs (
dict[str, Any], opsional) — Kamus kwargs yang jika ditentukan akan diteruskan keAttentionProcessorseperti yang didefinisikan di bawahself.processordalam diffusers.models.attention_processor. - return_dict (
bool, opsional, defaultTrue) — Apakah akan mengembalikan UNet2DConditionOutput alih-alih tuple biasa.
Metode forward dari DiTTransformer2DModel.