o
    pi*                     @   s  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ze	 s%e W n ey<   ddlmZ eee Y nw 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/< d0ged1< d2ged3< d4ged5< d6ged7< d8ged9< d:ged;< d<ged=< d>ged?< d@gedA< dBgedC< dDgedE< dFgedG< dHgedI< g dJedK< dLgedM< ze se W n ey   ddNlmZ eee Y n/w dOgedP< dQgedR< dSgedT< dUgedV< dWgedX< dYgedZ< d[ged\< d]ged^< g d_ed`< ze	 rJe sMe W n eye   ddalmZ eee Y nw dbgedc< ze	 rte
 swe W n ey   dddlmZ eee Y nw degedf< dggedh< eserEddilmZmZmZm	Z	m
Z
 z	e	 se W n ey   ddjlT Y nw dkdllmZmZ dkdmlmZ dkdnlmZ dkdolmZ dkdplmZ dkdqlmZ dkdrl m!Z! dkdsl"m#Z# dkdtl$m%Z% dkdul&m'Z' dkdvl(m)Z) dkdwl*m+Z+ dkdxl,m-Z- dkdyl.m/Z/ dkdzl0m1Z1 dkd{l2m3Z3 dkd|l4m5Z5 dkd}l6m7Z7 dkd~l8m9Z9 dkdl:m;Z; dkdl<m=Z= dkdl>m?Z? dkdl@mAZA dkdlBmCZC dkdlDmEZE dkdlFmGZG dkdlHmIZI dkdlJmKZK dkdlLmMZM dkdlNmOZO dkdlPmQZQ dkdlRmSZS dkdlTmUZU dkdlVmWZW dkdJlXmYZYmZZZm[Z[ dkdl\m]Z] z	e se W n ey   ddjl^T Y n=w dkdl_m`Z` dkdlambZb dkdlcmdZd dkdlemfZf dkdlgmhZh dkdlimjZj dkdlkmlZl dkdlmmnZn dkd_lompZpmqZqmrZrmsZs ze	 re se W n ey   ddjltT Y nw dkdlumvZv ze	 r#e
 s&e W n ey6   ddjlwT Y dS w dkdlxmyZy dkdlzm{Z{ dS d dl|Z|ee}e~ d eede|je}< e D ]\ZZee|je} ee q[dS )    )TYPE_CHECKING   )DIFFUSERS_SLOW_IMPORTOptionalDependencyNotAvailable_LazyModuleget_objects_from_moduleis_flax_availableis_scipy_availableis_torch_availableis_torchsde_available)dummy_pt_objectsKarrasVeSchedulerScoreSdeVpScheduler
deprecatedAmusedSchedulerscheduling_amusedConsistencyDecoderSchedulerscheduling_consistency_decoderCMStochasticIterativeSchedulerscheduling_consistency_modelsDDIMSchedulerscheduling_ddimCogVideoXDDIMSchedulerscheduling_ddim_cogvideoxDDIMInverseSchedulerscheduling_ddim_inverseDDIMParallelSchedulerscheduling_ddim_parallelDDPMSchedulerscheduling_ddpmDDPMParallelSchedulerscheduling_ddpm_parallelDDPMWuerstchenSchedulerscheduling_ddpm_wuerstchenDEISMultistepSchedulerscheduling_deis_multistepCogVideoXDPMSchedulerscheduling_dpm_cogvideox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FlowMatchHeunDiscreteScheduler#scheduling_flow_match_heun_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_flax_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_torch_and_scipy_objectsLMSDiscreteSchedulerscheduling_lms_discrete) dummy_torch_and_torchsde_objects!CosineDPMSolverMultistepScheduler%scheduling_cosine_dpmsolver_multistepDPMSolverSDESchedulerscheduling_dpmsolver_sde)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   )rL   )rN   )rP   )rV   )rY   )r[   )r]   )r_   )ra   )rc   )re   )rg   )ro   )rr   )rt   N__file__)module_spec)typingr   utilsr   r   r   r   r   r	   r
   r   _dummy_modules_import_structurer   updaterX   rn   rq   utils.dummy_pt_objectsr   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   rM   rL   rO   rN   rQ   rP   rU   rR   rS   rT   rW   rV   utils.dummy_flax_objectsrZ   rY   r\   r[   r^   r]   r`   r_   rb   ra   rd   rc   rf   re   rh   rg   rm   ri   rj   rk   rl   #utils.dummy_torch_and_scipy_objectsrp   ro   &utils.dummy_torch_and_torchsde_objectsrs   rr   ru   rt   sys__name__globals__spec__modulesitemsnamevaluesetattr r   r   [/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/schedulers/__init__.py<module>   sN  (












































