
    wi	(                     	   d dl mZ ddlmZmZmZmZmZmZm	Z	m
Z
 i Zi Z	  e	            s
 e            	 ddged<   dged<   d	ged
<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged<   dged <   d!ged"<   d#ged$<   d%ged&<   d'ged(<   d)ged*<   d+ged,<   d-ged.<   d/ged0<   d1ged2<   d3ged4<   d5ged6<   d7ged8<   d9ged:<   d;ged<<   d=ged><   d?ged@<   dAgedB<   g dCedD<   dEgedF<   n/# e$ r' ddGlmZ e                     ee                     Y nw xY w	  e            s
 e            	 dHgedI<   dJgedK<   dLgedM<   dNgedO<   dPgedQ<   dRgedS<   dTgedU<   dVgedW<   g dXedY<   n/# e$ r' ddZlmZ e                     ee                     Y nw xY w	  e	            r
 e            s
 e            	 d[ged\<   n/# e$ r' dd]lmZ e                     ee                     Y nw xY w	  e	            r
 e
            s
 e            	 d^ged_<   n/# e$ r' dd`lmZ e                     ee                     Y nw xY weserddalmZmZmZm	Z	m
Z
 	  e	            s
 e            	 dbdclmZmZ dbddlmZ dbdelmZ dbdflmZ dbdglmZ dbdhlmZ dbdilm Z  dbdjl!m"Z" dbdkl#m$Z$ dbdll%m&Z& dbdml'm(Z( dbdnl)m*Z* dbdol+m,Z, dbdpl-m.Z. dbdql/m0Z0 dbdrl1m2Z2 dbdsl3m4Z4 dbdtl5m6Z6 dbdul7m8Z8 dbdvl9m:Z: dbdwl;m<Z< dbdxl=m>Z> dbdyl?m@Z@ dbdzlAmBZB dbd{lCmDZD dbd|lEmFZF dbd}lGmHZH dbd~lImJZJ dbdlKmLZL dbdlMmNZN dbdlOmPZP dbdClQmRZRmSZSmTZT dbdlUmVZV n# e$ r ddlWT Y nw xY w	  e            s
 e            	 dbdlXmYZY dbdlZm[Z[ dbdl\m]Z] dbdl^m_Z_ dbdl`maZa dbdlbmcZc dbdldmeZe dbdlfmgZg dbdXlhmiZimjZjmkZkmlZl n# e$ r ddlmT Y nw xY w	  e	            r
 e            s
 e            	 dbdlnmoZo n# e$ r ddlpT Y nw xY w	  e	            r
 e
            s
 e            	 dbdlqmrZr dS # e$ r ddlsT Y dS w xY wd dltZt eeu ev            d         eew          etjx        eu<   ey                                D ]\  ZzZ{ e|etjx        eu         eze{           dS )    )TYPE_CHECKING   )DIFFUSERS_SLOW_IMPORTOptionalDependencyNotAvailable_LazyModuleget_objects_from_moduleis_flax_availableis_scipy_availableis_torch_availableis_torchsde_availableKarrasVeSchedulerScoreSdeVpScheduler
deprecatedAmusedSchedulerscheduling_amusedConsistencyDecoderSchedulerscheduling_consistency_decoderCMStochasticIterativeSchedulerscheduling_consistency_modelsDDIMSchedulerscheduling_ddimDDIMInverseSchedulerscheduling_ddim_inverseDDIMParallelSchedulerscheduling_ddim_parallelDDPMSchedulerscheduling_ddpmDDPMParallelSchedulerscheduling_ddpm_parallelDDPMWuerstchenSchedulerscheduling_ddpm_wuerstchenDEISMultistepSchedulerscheduling_deis_multistepDPMSolverMultistepSchedulerscheduling_dpmsolver_multistep"DPMSolverMultistepInverseScheduler&scheduling_dpmsolver_multistep_inverseDPMSolverSinglestepSchedulerscheduling_dpmsolver_singlestepEDMDPMSolverMultistepScheduler"scheduling_edm_dpmsolver_multistepEDMEulerSchedulerscheduling_edm_eulerEulerAncestralDiscreteScheduler#scheduling_euler_ancestral_discreteEulerDiscreteSchedulerscheduling_euler_discreteFlowMatchEulerDiscreteScheduler$scheduling_flow_match_euler_discreteHeunDiscreteSchedulerscheduling_heun_discreteIPNDMSchedulerscheduling_ipndmKDPM2AncestralDiscreteScheduler%scheduling_k_dpm_2_ancestral_discreteKDPM2DiscreteSchedulerscheduling_k_dpm_2_discreteLCMSchedulerscheduling_lcmPNDMSchedulerscheduling_pndmRePaintSchedulerscheduling_repaintSASolverSchedulerscheduling_sasolverScoreSdeVeSchedulerscheduling_sde_veTCDSchedulerscheduling_tcdUnCLIPSchedulerscheduling_unclipUniPCMultistepSchedulerscheduling_unipc_multistep)AysSchedulesKarrasDiffusionSchedulersSchedulerMixinscheduling_utilsVQDiffusionSchedulerscheduling_vq_diffusion)dummy_pt_objectsFlaxDDIMSchedulerscheduling_ddim_flaxFlaxDDPMSchedulerscheduling_ddpm_flaxFlaxDPMSolverMultistepScheduler#scheduling_dpmsolver_multistep_flaxFlaxEulerDiscreteSchedulerscheduling_euler_discrete_flaxFlaxKarrasVeSchedulerscheduling_karras_ve_flaxFlaxLMSDiscreteSchedulerscheduling_lms_discrete_flaxFlaxPNDMSchedulerscheduling_pndm_flaxFlaxScoreSdeVeSchedulerscheduling_sde_ve_flax)FlaxKarrasDiffusionSchedulersFlaxSchedulerMixinFlaxSchedulerOutputbroadcast_to_shape_from_leftscheduling_utils_flax)dummy_flax_objectsLMSDiscreteSchedulerscheduling_lms_discrete)dummy_torch_and_scipy_objectsDPMSolverSDESchedulerscheduling_dpmsolver_sde) dummy_torch_and_torchsde_objects)r   r	   r
   r   r      )r   r   )r   )r   )r   )r   )r   )r   )r   )r   )r    )r"   )r$   )r&   )r(   )r*   )r,   )r.   )r0   )r2   )r4   )r6   )r8   )r:   )r<   )r>   )r@   )rB   )rD   )rF   )rH   )rJ   )rP   )*)rS   )rU   )rW   )rY   )r[   )r]   )r_   )ra   )ri   )rl   N__file__)module_spec)}typingr   utilsr   r   r   r   r	   r
   r   r   _dummy_modules_import_structurerR   updaterh   rk   rn   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r!   r    r#   r"   r%   r$   r'   r&   r)   r(   r+   r*   r-   r,   r/   r.   r1   r0   r3   r2   r5   r4   r7   r6   r9   r8   r;   r:   r=   r<   r?   r>   rA   r@   rC   rB   rE   rD   rG   rF   rI   rH   rK   rJ   rO   rL   rM   rN   rQ   rP   utils.dummy_pt_objectsrT   rS   rV   rU   rX   rW   rZ   rY   r\   r[   r^   r]   r`   r_   rb   ra   rg   rc   rd   re   rf   utils.dummy_flax_objectsrj   ri   #utils.dummy_torch_and_scipy_objectsrm   rl   &utils.dummy_torch_and_torchsde_objectssys__name__globals__spec__modulesitemsnamevaluesetattr     m/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/schedulers/__init__.py<module>r      sP   !          	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	  )L /,,.../ (;<Q&Rl#.?-@)*;X:Y67:Z9[56,;+<'(4J3K/05L4M01,;+<'(5L4M017P6Q236N5O12;X:Y67CgBh>?<Z;[78?_>`:;1D0E,-@a?b;<6N5O12Ab@c<=5L4M01-=,>()BcAd=>8P7Q34+9*:&',;+<'(/A.B*+0C/D+,.C-D)*+9*:&'.?-@)*7P6Q23,k,k,k()4J3K/00M & E E E((((((112BCCDDDDDEP /,,.../ 2E0E,-1D0E,-@a?b;<;W:X676M5N129S8T451D0E,-3L2M./2 2 2-.. & G G G******112DEEFFFFFG,	L   /%7%7%9%9 /,,.../ 5K3K/00 & R R R555555112OPPQQQQQR	N   /%:%:%<%< /,,.../ 6M4M011 & U U U888888112RSSTTTTTU  [4) [4             &B!!## 	300222	3
 	GFFFFFFF666666OOOOOOQQQQQQ222222AAAAAACCCCCC222222CCCCCCGGGGGGEEEEEEOOOOOO^^^^^^QQQQQQVVVVVV;;;;;;XXXXXXEEEEEEYYYYYYCCCCCC444444ZZZZZZGGGGGG000000222222888888::::::::::::000000666666GGGGGG]]]]]]]]]]AAAAAAAG * / / /....../J
  "" 	300222	3
 	<;;;;;;;;;;;XXXXXXNNNNNNDDDDDDJJJJJJ;;;;;;CCCCCC	
 	
 	
 	
 	
 	
 	
 	
 	
 	
 	
 	
 	
 * 1 1 10000001$B""$$ 	3););)=)= 	300222	3
 	BAAAAAA * < < <;;;;;;<
D""$$ 	3)>)>)@)@ 	300222	3
 	DCCCCCCC * ? ? ?>>>>>>>? JJJ'K''))J2GIZhpqqqCK%++-- 4 4eH%tU33334 4s   C> >)D*)D*.E; ;)F'&F'+G )G=<G=H' ')II*M 	MMN. .	N:9N:>O$ $	O0/O04P 	P('P(