o
    *jU                     @   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stepsZddimi  Zlinear)Zuse_timestepsbetasZmodel_mean_typeZmodel_var_type	loss_typerescale_timesteps)gdZget_named_beta_scheduleZLossTypeZMSEr   r   ZModelMeanTypeZEPSILONZSTART_XZModelVarTypeZFIXED_LARGEZFIXED_SMALLZLEARNED_RANGE)Zdiffusion_configZpredict_xstartZsigma_smallZlearn_sigmar   Ztimestep_respacingZdiffusion_stepsr	   r   r   Z	diffusion r   v/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/multi_modal/guided_diffusion/script.pycreate_diffusion   s6   


r   N)Z6modelscope.models.cv.motion_generation.modules.respacer    r   r
   Zrespacer   r   r   r   r   r   <module>   s   