o
    7i                     @   s8   d dl mZ d dlmZmZ 							d
ddZd	S )   )gaussian_diffusion)SpacedDiffusionspace_timestepslinearFT  c           
      C   s   t ||}|rt jj}	n|rt jj}	nt jj}	| d u s | dkr#|g} tt|| ||s/t jj	nt jj
|sD|s=t jj|	dS t jj|	dS t jj|	dS )N )Zuse_timestepsbetasZmodel_mean_typeZmodel_var_type	loss_type)gdZget_named_beta_scheduleZLossTypeZRESCALED_KLZRESCALED_MSEZMSEr   r   ZModelMeanTypeEPSILONZSTART_XZModelVarTypeZFIXED_LARGEZFIXED_SMALLZLEARNED_RANGE)
timestep_respacingnoise_scheduleZuse_klZsigma_smallZpredict_xstartZlearn_sigmaZrescale_learned_sigmasZdiffusion_stepsr   r	    r   L/data/cameron/vidgen/unified_video_action/./simple_uva/diffusion/__init__.pycreate_diffusion   s0   



r   N)r   FFFTFr   )r   r   r
   Zrespacer   r   r   r   r   r   r   <module>   s   