o
    ߥiU                     @   s0   d dl mZ ddlmZ ddlmZ dd ZdS )    )space_timesteps   )gaussian_diffusion)SpacedDiffusionc                 C   s   d}d}d}| d }d| }d}d}t d|}t jj}	|s"|g}tt||||s.t jjnt jj|s=|s9t j	j
nt j	jnt j	j|	|d}
|
S )NFTstepsddimi  linear)use_timestepsbetasmodel_mean_typemodel_var_type	loss_typerescale_timesteps)gdget_named_beta_scheduleLossTypeMSEr   r   ModelMeanTypeEPSILONSTART_XModelVarTypeFIXED_LARGEFIXED_SMALLLEARNED_RANGE)diffusion_configpredict_xstartsigma_smalllearn_sigmar   timestep_respacingdiffusion_stepsr   r
   r   	diffusion r!   i/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/multi_modal/guided_diffusion/script.pycreate_diffusion   s6   


r#   N)6modelscope.models.cv.motion_generation.modules.respacer    r   r   respacer   r#   r!   r!   r!   r"   <module>   s   