Diffusers documentation

LuminaNextDiT2DModel

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

LuminaNextDiT2DModel

نسخة جديدة من نموذج Diffusion Transformer للبيانات ثنائية الأبعاد من Lumina-T2X.

LuminaNextDiT2DModel

فئة diffusers.LuminaNextDiT2DModel

< >

( sample_size: int = 128patch_size: int | None = 2in_channels: int | None = 4hidden_size: int | None = 2304num_layers: int | None = 32num_attention_heads: int | None = 32num_kv_heads: int | None = Nonemultiple_of: int | None = 256ffn_dim_multiplier: float | None = Nonenorm_eps: float | None = 1e-05learn_sigma: bool | None = Trueqk_norm: bool | None = Truecross_attention_dim: int | None = 2048scaling_factor: float | None = 1.0 )

المعاملات

  • sample_size (int) — عرض الصور الكامنة. هذا ثابت أثناء التدريب لأنه يُستخدم لتعلم عدد من تضمينات المواقع.
  • patch_size (int, اختياري, (int, اختياري, الافتراضي 2) — حجم كل رقعة في الصورة. يحدد هذا المعامل دقة الرقع المُدخلة إلى النموذج.
  • in_channels (int، اختياري، القيمة الافتراضية 4) — عدد قنوات الإدخال للنموذج. عادةً، يتطابق هذا العدد مع عدد القنوات في صور الإدخال.
  • hidden_size (int، اختياري، القيمة الافتراضية 4096) — أبعاد الطبقات المخفية في النموذج. يحدد هذا المعامل عرض التمثيلات المخفية للنموذج.
  • num_layers (int، اختياري، الافتراضي 32) — عدد الطبقات في النموذج. يحدد هذا عمق الشبكة العصبية.
  • num_attention_heads (int، اختياري، القيمة الافتراضية 32) — عدد رؤوس الانتباه في كل طبقة انتباه. يحدد هذا المعامل عدد آليات الانتباه المنفصلة المستخدمة.
  • num_kv_heads (int، اختياري، القيمة الافتراضية 8) — عدد رؤوس المفتاح-القيمة في آلية الانتباه، إذا كان مختلفًا عن عدد رؤوس الانتباه. إذا كانت القيمة None، فإنها تتحول افتراضيًا إلى num_attention_heads.
  • multiple_of (int، اختياري، القيمة الافتراضية 256) — عامل يجب أن يكون حجم المخفي مضاعفًا له. يمكن أن يساعد ذلك في تحسين بعض تكوينات الأجهزة.
  • ffn_dim_multiplier (float, اختياري) — معامل مضاعف لأبعاد الشبكة الأمامية. إذا كانت القيمة None، يتم استخدام قيمة افتراضية بناءً على إعدادات النموذج.
  • norm_eps (float، اختياري، القيمة الافتراضية 1e-5) — قيمة صغيرة تُضاف إلى المقام لضمان الاستقرار العددي في طبقات التطبيع.
  • learn_sigma (bool، اختياري، القيمة الافتراضية True) — ما إذا كان يجب على النموذج تعلّم معامل sigma، والذي قد يكون مرتبطًا بعدم اليقين أو التباين في التنبؤات.
  • qk_norm (bool، اختياري، القيمة الافتراضية True) — يحدد ما إذا كان ينبغي تطبيع الاستعلامات والمفاتيح في آلية الانتباه.
  • cross_attention_dim (int، اختياري، القيمة الافتراضية 2048) — أبعاد تمثيلات النص. يحدد هذا المعامل حجم التمثيلات النصية المستخدمة في النموذج.
  • scaling_factor (float, اختياري، القيمة الافتراضية 1.0) — عامل مقياس يُطبَّق على معلمات أو طبقات معينة في النموذج. يمكن استخدامه لضبط المقياس العام لعمليات النموذج.

LuminaNextDiT: نموذج انتشار بهيكل أساسي من نوع Transformer.

ورث ModelMixin وConfigMixin لتكون متوافقة مع عينة StableDiffusionPipeline من diffusers.

forward

< >

( hidden_states: Tensortimestep: Tensorencoder_hidden_states: Tensorencoder_mask: Tensorimage_rotary_emb: Tensorcross_attention_kwargs: dict = Nonereturn_dict = True ) → ~models.transformer_2d.Transformer2DModelOutput or tuple

المعاملات

  • hidden_states (torch.Tensor) — متجه إدخال على شكل (N, C, H, W).
  • timestep (torch.Tensor) — متجه (Tensor) لخطوات الانتشار (diffusion timesteps) من الشكل (N,).
  • encoder_hidden_states (torch.Tensor) — متجه لميزات العنوان (caption) بأبعاد (N, D).
  • encoder_mask (torch.Tensor) — متجه لأقنعة العناوين ذات الشكل (N, L).
  • image_rotary_emb (torch.Tensor) — تضمينات موضعية دوّارة محسوبة مسبقًا.
  • cross_attention_kwargs (dict، اختياري) — قاموس kwargs يُمرَّر إلى AttentionProcessor كما هو معرَّف تحت self.processor في diffusers.models.attention_processor إذا تمَّ تحديده.
  • return_dict (bool، اختياري، القيمة الافتراضية True) — ما إذا كان يجب إرجاع ~models.transformer_2d.Transformer2DModelOutput بدلاً من tuple عادي.

يعيد

~models.transformer_2d.Transformer2DModelOutput أو tuple

إذا كان return_dict صحيحًا، يُعاد ~models.transformer_2d.Transformer2DModelOutput، وإلا فإنه يُعاد tuple عادي.

مرور أمامي في LuminaNextDiT.

Update on GitHub