Hugging Face's logo

Diffusers documentation

HeliosDMDScheduler

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

HeliosDMDScheduler

HeliosDMDScheduler مبني على أخذ العينات المطابق للتدفق الهرمي الذي تم تقديمه في Helios.

HeliosDMDScheduler

فئة diffusers.HeliosDMDScheduler

< >

( num_train_timesteps: int = 1000shift: float = 1.0stages: int = 3stage_range: list = [0, 0.3333333333333333, 0.6666666666666666, 1]gamma: float = 0.3333333333333333prediction_type: str = 'flow_prediction'use_flow_sigmas: bool = Trueuse_dynamic_shifting: bool = Falsetime_shift_type: typing.Literal['exponential', 'linear'] = 'linear' )

init_sigmas

< >

( )

تهيئة الخطوات الزمنية العالمية والقيم السيجما

init_sigmas_for_each_stage

< >

( )

تهيئة خطوات الزمن لكل مرحلة

set_begin_index

< >

( begin_index: int = 0 )

المعاملات

  • begin_index (int) — مؤشر البداية للمجدول.

يحدد الفهرس الابتدائي للمجدول. يجب تشغيل هذه الدالة من خط الأنابيب قبل الاستدلال.

set_timesteps

< >

( num_inference_steps: intstage_index: int | None = Nonedevice: typing.Union[str, torch.device] = Nonesigmas: bool | None = Nonemu: bool | None = Noneis_amplify_first_chunk: bool = False )

ضبط خطوات الزمن والقيم السيجما لكل مرحلة

time_shift

< >

( mu: floatsigma: floatt: Tensor ) → torch.Tensor

المعاملات

  • mu (float) — معامل mu الخاص بإزاحة الوقت.
  • sigma (float) — معامل sigma الخاص بإزاحة الوقت.
  • t (torch.Tensor) — خطوات الإدخال الزمنية.

يعيد

torch.Tensor

خطوات الزمن المُزاحة.

طبّق إزاحةً زمنية على قيم sigmas.

scheduling_helios_dmd

Update on GitHub

←FlowMatchHeunDiscreteScheduler HeliosScheduler→