Buckets:
AutoencoderKLHunyuanImage
نموذج الترميز التلقائي التبايني ثنائي الأبعاد (VAE) مع خسارة كولباك-ليبلر المستخدم في [HunyuanImage2.1].
يمكن تحميل النموذج باستخدام مقتطف الشيفرة التالي.
from diffusers import AutoencoderKLHunyuanImage
vae = AutoencoderKLHunyuanImage.from_pretrained("hunyuanvideo-community/HunyuanImage-2.1-Diffusers", subfolder="vae", dtype=torch.bfloat16)
AutoencoderKLHunyuanImage[[diffusers.AutoencoderKLHunyuanImage]]
diffusers.AutoencoderKLHunyuanImage[[diffusers.AutoencoderKLHunyuanImage]]
diffusers.AutoencoderKLHunyuanImage(in_channels: int, out_channels: int, latent_channels: int, block_out_channels: tuple[int, ...], layers_per_block: int, spatial_compression_ratio: int, sample_size: int, scaling_factor: float = None, downsample_match_channel: bool = True, upsample_match_channel: bool = True)
نموذج VAE للصور ثنائية الأبعاد مع دعم التقطيع المكاني.
يرث هذا النموذج من ModelMixin. راجع توثيق الفئة الأصلية لمعرفة طرقه العامة المُنفَّذة لجميع النماذج (مثل التنزيل أو الحفظ).
فك الترميز[[diffusers.AutoencoderKLHunyuanImage.decode]]
decode(z: Tensor, return_dict: bool = True)
المعاملات:
z (torch.Tensor) : دفعة إدخال من المتجهات الكامنة.
return_dict (bool، اختياري، القيمة الافتراضية True) : ما إذا كان يجب إرجاع ~models.vae.DecoderOutput بدلاً من tuple عادي.
يعيد: ~models.vae.DecoderOutput أو tuple
إذا كانت return_dict تساوي True، يُعاد ~models.vae.DecoderOutput، وإلا يُعاد tuple عادي.
فكّ تشفير دفعة من الصور.
enable_tiling[[diffusers.AutoencoderKLHunyuanImage.enable_tiling]]
enable_tiling(tile_sample_min_size: int | None = None, tile_overlap_factor: float | None = None)
المعاملات:
tile_sample_min_size (int، اختياري) : الحد الأدنى للحجم المطلوب لعينة حتى يتم فصلها إلى بلاطات عبر البُعد المكاني.
tile_overlap_factor (float، اختياري) : عامل التداخل المطلوب لفصل كامن إلى بلاطات عبر البُعد المكاني.
تفعيل فك ترميز VAE المُقسّم مكانيًا. عند تفعيل هذا الخيار، سيقوم VAE بتقسيم المصفوفة المدخلة إلى بلاطات (tiles) لحساب فك الترميز والترميز على عدة خطوات. وهذا مفيد لتوفير كمية كبيرة من الذاكرة ولتيسير معالجة صور أكبر.
ترميز[[diffusers.AutoencoderKLHunyuanImage.encode]]
encode(x: Tensor, return_dict: bool = True)
المعاملات:
x (torch.Tensor) : دفعة من الصور المدخلة.
return_dict (bool، اختياري، الافتراضي True) : ما إذا كان يجب إرجاع ~models.autoencoder_kl.AutoencoderKLOutput بدلاً من tuple عادي.
يعيد:
تمثيلات الكامنة للفيديوهات المُرمّزة. إذا كان return_dict صحيحًا (True)، يُعاد ~models.autoencoder_kl.AutoencoderKLOutput، وإلا يُعاد tuple عادي.
ترميز دفعة من الصور إلى كامنات.
forward[[diffusers.AutoencoderKLHunyuanImage.forward]]
forward(sample: Tensor, sample_posterior: bool = False, return_dict: bool = True, generator: typing.Optional[torch.Generator] = None)
المعاملات:
sample (torch.Tensor) : عينة الإدخال.
sample_posterior (bool، اختياري، الافتراضي False) : ما إذا كان سيتم أخذ عينة من التوزيع اللاحق.
return_dict (bool، اختياري، الافتراضي True) : ما إذا كان يجب إرجاع DecoderOutput بدلاً من tuple عادي.
generator (torch.Generator، اختياري) : torch.Generator لجعل عملية أخذ العينات حتمية.
يعيد: ~models.vae.DecoderOutput أو tuple
إذا كان return_dict صحيحًا، يُعاد ~models.vae.DecoderOutput، وإلا يُعاد tuple عادي.
tiled_decode[[diffusers.AutoencoderKLHunyuanImage.tiled_decode]]
tiled_decode(z: Tensor, return_dict: bool = True)
المعاملات:
z (torch.Tensor) : متجه كامن (Latent tensor) على شكل (B, C, H, W).
return_dict (bool، اختياري، القيمة الافتراضية True) : ما إذا كان يجب إرجاع ~models.vae.DecoderOutput بدلاً من tuple عادي.
يعيد: ~models.vae.DecoderOutput أو tuple
إذا كانت return_dict تساوي True، يُعاد ~models.vae.DecoderOutput، وإلا يُعاد tuple عادي.
فكّ الترميز الكامن باستخدام استراتيجية التقطيع المكاني.
tiled_encode[[diffusers.AutoencoderKLHunyuanImage.tiled_encode]]
tiled_encode(x: Tensor)
المعاملات:
x (torch.Tensor) : متجه الإدخال ذو الشكل (B, C, T, H, W).
يعيد: torch.Tensor
التمثيل الكامن للصور المُرمّزة.
ترميز المدخلات باستخدام استراتيجية التقطيع المكاني.
DecoderOutput[[diffusers.models.autoencoders.vae.DecoderOutput]]
diffusers.models.autoencoders.vae.DecoderOutput[[diffusers.models.autoencoders.vae.DecoderOutput]]
diffusers.models.autoencoders.vae.DecoderOutput(sample: Tensor, commit_loss: typing.Optional[torch.FloatTensor] = None)
المعاملات:
sample (torch.Tensor من الشكل (batch_size, num_channels, height, width)) : عينة الإخراج المُفكّكة من الطبقة الأخيرة للنموذج.
مخرجات طريقة فك الترميز.
Xet Storage Details
- Size:
- 7.36 kB
- Xet hash:
- ab1b52e94d77bce3ae99d37a5567f08fa5865e2ae1211b7c630825fb221139dc
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.