o
    *j                     @   s  d dl Z d dlZd dlmZ d dlmZ d dlZd dlmZ eddZ	e
eddZej Zee Zg de	_d	e	_d
e	_de	_de	_de	_de	_de	_d
e	_de	_g de	_g de	_g de	_g de	_de	_g de	_ dge	_!de	_"e#dde#dde#dde#ddie	_$de	_%de	_&de	_'de	_(de	_)de	_*de	_+de	_,de	_-de	_.d e	_/de	_0de	_1d!e	_2d"e	_3d#e	_4d$e	_5d%e	_6d&e	_7d'e	_8de	_9d(e	_:g d)e	_;g d*e	_<d+e	_=de	_>de	_?de	_@e	j8e	_Ad&e	_Be	j+Cd,rdnde	_Dd-e	_Eg d.e	_Fde	_Gde	_Hde	_Ig d/e	_Jd0e	_Kd1e	_Ld0e	_Md0e	_Nde	_Ode	_Pd2e	_Qde	_Rde	_Sde	_Tde	_Ud2e	_Vd3e	_Wd4e	_XejYe	jXd5e	jW d6e	jW d7e	_Zde	_[e	j[r[ejYe	jXd8e	jW d9e	_[d2e	_\edk rfde	_\de	_]d2e	_^de	__de	_`d2e	_ad:e	_bde	_cde	_dd;e	_ed<Ye	jZfd=e	_gdS )>    N)datetime)EasyDictzConfig: VideoComposer)__name__Z
WORLD_SIZE   )textmaskZdepthmapZsketchZmotionimageZlocal_imageZsingle_sketchz
webvid10m/gffffff?i  g      4@g      $@F   )      ?r
   r
   )g
ףp=
?gv/?gCl?)gZd;O?gy&1?g?i  )   r   r   r         r   @      i"  epsZfixed_smallZmse2   g              ?g?g      @zv2-1_512-ema-pruned.ckptiP     i   i      )g3<4'?gwgM?gy{ ?)gB91?gwt.?g	U?zopen_clip_pytorch_model.binZlearnedi@  )r   r   r   r   )r   r
   g      ?g?r
   Ti B .Zstep_z	/non_ema_z.pthZoptimizer_step_z.ptgH.?d   _zoutputs/)hloggingosZos.pathpathZospr   ZtorchZeasydictr   cfgintgetenvZpmi_world_sizecudaZdevice_countZgpus_per_machineZ
world_sizeZvideo_compositionsroot_diralphaZ	misc_sizeZ	depth_stdZdepth_clampZ
hist_sigmaZuse_image_datasetZ	alpha_img
resolutionmeanZstdZsketch_meanZ
sketch_stdZ	max_wordsZ
frame_lensZfeature_frameratesZfeature_frameratestrZbatch_sizes
chunk_sizeZnum_workersZprefetch_factorseedZnum_timestepsZ	mean_typeZvar_typeZ	loss_typeZddim_timestepsZddim_etaclampZshare_noiseZuse_div_lossZp_zeroZguide_scaleZsd_checkpointZvit_image_sizeZvit_patch_sizeZvit_dimZvit_out_dimZ	vit_headsZ
vit_layersZvit_meanZvit_stdZclip_checkpointZ
mvs_visualZunet_in_dimZunet_concat_dimZ
unet_y_dimZunet_context_dim
startswithZunet_out_dimZunet_dimZunet_dim_multZunet_res_blocksZunet_num_headsZunet_head_dimZunet_attn_scalesZunet_dropoutZmisc_dropoutZ
p_all_zeroZ
p_all_keepZtemporal_convZtemporal_attn_timesZtemporal_attentionZuse_fps_conditionZuse_sim_maskZ
pretrainedZ
fix_weightresumeZresume_stepZresume_check_dirjoinZresume_checkpointZresume_optimizerZuse_emaZ	load_fromZuse_checkpointZuse_sharded_ddpZuse_fsdpZuse_fp16Z	ema_decayZviz_intervalZsave_ckp_intervalZlog_intervalZcomposition_stringsZlog_dir r*   r*   s/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/multi_modal/videocomposer/config.py<module>   s    













