o
    ߥi                     @   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__
WORLD_SIZE   )textmaskdepthmapsketchmotionimagelocal_image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"  epsfixed_small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.binlearnedi@  )r   r   r   r   )r   r   g      ?g?r   Ti B .step_z	/non_ema_z.pthoptimizer_step_z.ptgH.?d   _zoutputs/)hloggingosos.pathpathospr   torcheasydictr   cfgintgetenvpmi_world_sizecudadevice_countgpus_per_machine
world_sizevideo_compositionsroot_diralpha	misc_size	depth_stddepth_clamp
hist_sigmause_image_dataset	alpha_img
resolutionmeanstdsketch_mean
sketch_std	max_words
frame_lensfeature_frameratesfeature_frameratestrbatch_sizes
chunk_sizenum_workersprefetch_factorseednum_timesteps	mean_typevar_type	loss_typeddim_timestepsddim_etaclampshare_noiseuse_div_lossp_zeroguide_scalesd_checkpointvit_image_sizevit_patch_sizevit_dimvit_out_dim	vit_heads
vit_layersvit_meanvit_stdclip_checkpoint
mvs_visualunet_in_dimunet_concat_dim
unet_y_dimunet_context_dim
startswithunet_out_dimunet_dimunet_dim_multunet_res_blocksunet_num_headsunet_head_dimunet_attn_scalesunet_dropoutmisc_dropout
p_all_zero
p_all_keeptemporal_convtemporal_attn_timestemporal_attentionuse_fps_conditionuse_sim_mask
pretrained
fix_weightresumeresume_stepresume_check_dirjoinresume_checkpointresume_optimizeruse_ema	load_fromuse_checkpointuse_sharded_ddpuse_fsdpuse_fp16	ema_decayviz_intervalsave_ckp_intervallog_intervalcomposition_stringslog_dir r   r   f/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/multi_modal/videocomposer/config.py<module>   s    













