Buckets:

hf-doc-build/doc / diffusers /main /ar /api /models /chroma_transformer.md
mishig's picture
|
download
raw
5.54 kB
# ChromaTransformer2DModel
نموذج flux Transformer معدّل من [Chroma](https://huggingface.co/lodestones/Chroma1-HD)
## ChromaTransformer2DModel[[diffusers.ChromaTransformer2DModel]]
#### diffusers.ChromaTransformer2DModel[[diffusers.ChromaTransformer2DModel]]
```python
diffusers.ChromaTransformer2DModel(patch_size: int = 1, in_channels: int = 64, out_channels: int | None = None, num_layers: int = 19, num_single_layers: int = 38, attention_head_dim: int = 128, num_attention_heads: int = 24, joint_attention_dim: int = 4096, axes_dims_rope: tuple[int, ...] = (16, 56, 56), approximator_num_channels: int = 64, approximator_hidden_dim: int = 5120, approximator_layers: int = 5)
```
[المصدر](https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/transformers/transformer_chroma.py#L359)
**المعاملات:**
patch_size (`int`، القيمة الافتراضية `1`) : حجم الباتش لتحويل البيانات المدخلة إلى باتشات صغيرة.
in_channels (`int`، القيمة الافتراضية `64`) : عدد القنوات في المدخلات.
out_channels (`int`، *اختياري*، القيمة الافتراضية `None`) : عدد القنوات في المخرجات. إذا لم يُحدد، تكون القيمة الافتراضية `in_channels`.
num_layers (`int`، القيمة الافتراضية `19`) : عدد طبقات كتل DiT ذات التدفق المزدوج التي سيتم استخدامها.
num_single_layers (`int`، القيمة الافتراضية `38`) : عدد طبقات كتل DiT أحادية التدفق المستخدمة.
attention_head_dim (`int`، القيمة الافتراضية `128`) : عدد الأبعاد المستخدمة لكل رأس انتباه.
num_attention_heads (`int`، القيمة الافتراضية `24`) : عدد رؤوس الانتباه (attention heads) المستخدمة.
joint_attention_dim (`int`، القيمة الافتراضية `4096`) : عدد الأبعاد المستخدمة للانتباه المشترك (بعد التضمين/القناة الخاص بـ `encoder_hidden_states`).
axes_dims_rope (`tuple[int]`، القيمة الافتراضية هي `(16, 56, 56)`) : الأبعاد المستخدمة للتضمينات الموضعية الدوارة.
نموذج Transformer الذي تم تقديمه في Flux، مع تعديلات من أجل Chroma.
مرجع: https://huggingface.co/lodestones/Chroma1-HD
#### forward[[diffusers.ChromaTransformer2DModel.forward]]
```python
forward(hidden_states: Tensor, encoder_hidden_states: Tensor = None, timestep: LongTensor = None, img_ids: Tensor = None, txt_ids: Tensor = None, attention_mask: Tensor = None, joint_attention_kwargs: dict[str, typing.Any] | None = None, controlnet_block_samples = None, controlnet_single_block_samples = None, return_dict: bool = True, controlnet_blocks_repeat: bool = False)
```
[المصدر](https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/transformers/transformer_chroma.py#L465)
**المعاملات:**
hidden_states (`torch.Tensor` من الشكل `(batch_size, image_sequence_length, in_channels)`) : مدخلات `hidden_states`.
encoder_hidden_states (`torch.Tensor` من الشكل `(batch_size, text_sequence_length, joint_attention_dim)`) : تمثيلات شرطية (تمثيلات محسوبة من الشروط المدخلة مثل الأوامر النصية) للاستخدام.
timestep ( `torch.LongTensor`) : يُستخدم للإشارة إلى خطوة إزالة الضوضاء.
img_ids (`torch.Tensor`) : معرّفات مواضع الصور المستخدمة لحساب التضمينات الموضعية الدوارة.
txt_ids (`torch.Tensor`) : معرّفات مواضع النص المستخدمة لحساب التضمينات الموضعية الدوارة.
attention_mask (`torch.Tensor`، *اختياري*) : القناع المطبَّق على `encoder_hidden_states` أثناء الانتباه.
controlnet_block_samples (`list` من `torch.Tensor`، *اختياري*) : قائمة من الموترات التي إذا تم تحديدها تُضاف إلى بقايا كتل التحويل.
controlnet_single_block_samples (`list` من `torch.Tensor`، *اختياري*) : قائمة من الموترات التي إذا تم تحديدها تُضاف إلى بقايا كتل التحويل المفردة.
controlnet_blocks_repeat (`bool`، *اختياري*، القيمة الافتراضية `False`) : ما إذا كان يجب تكرار عينات كتلة controlnet عبر جميع كتل transformer.
joint_attention_kwargs (`dict`، *اختياري*) : قاموس kwargs يُمرَّر إلى `AttentionProcessor` إذا تم تحديده، كما هو معرَّف تحت `self.processor` في [diffusers.models.attention_processor](https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/attention_processor.py).
return_dict (`bool`، *اختياري*، القيمة الافتراضية `True`) : ما إذا كان يجب إرجاع `~models.transformer_2d.Transformer2DModelOutput` بدلاً من tuple عادي.
**يعيد:**
إذا كان `return_dict` صحيحًا، يُعاد `~models.transformer_2d.Transformer2DModelOutput`، وإلا يُعاد `tuple` يكون عنصره الأول هو تينسور العينة.
طريقة [FluxTransformer2DModel](/docs/diffusers/main/ar/api/models/flux_transformer#diffusers.FluxTransformer2DModel) الأمامية.

Xet Storage Details

Size:
5.54 kB
·
Xet hash:
6a6fad606f898d2154ca78fe6de96c9b851a1f73b4d1b0bece36a68d92837f5e

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.