Diffusers documentation
LuminaNextDiT2DModel
LuminaNextDiT2DModel
نسخة جديدة من نموذج Diffusion Transformer للبيانات ثنائية الأبعاد من Lumina-T2X.
LuminaNextDiT2DModel
فئة diffusers.LuminaNextDiT2DModel
< source >( 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
< source >( 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.