o g)@s*ddlTddlZddlZGdddZdS))*Nc@s0eZdZd ddZ        d dd ZdS) merge_config_then_runreturnNcCsd}d|_d|_d|_d|_d}|rAtj|ddd|_tj|dd|_tj|dd|_d d gd d }tj t j |d |d|_dSdS)Nz#FateZero/ckpt/stable-diffusion-v1-4T tokenizerF) subfolderZuse_fast text_encoder)rvaemidi)ZloraZSparseCausalAttention_indexZleast_sc_channelunet) model_config) rrrr Z AutoTokenizerZfrom_pretrainedZ CLIPTextModelZ AutoencoderKLZUNetPseudo3DConditionModelZ from_2d_modelospathjoin)selfZpretrained_model_pathZ cache_ckptr r5/data/xianyang/code/VideoGrain/merge_config_gradio.py__init__s8 zmerge_config_then_run.__init__rcCsd}t|}t}t|}td|||dd<||dd<||||d}| | ||d}|d|| rB|durBtd| durct | t rR| |dd<nt | d rc| j durc| j |dd<|g|d d <t|d d d }||dd<||d<| d|dd<|gt|dd|dd<t||d d d <| |d d<| |d d<|ddddddd|}||d<td|td"|j|j|j|j|d|}|d d!}|S)#NzFFateZero/config/low_resource_teaser/jeep_watercolor_ddim_10_steps.yamlz/config_now['pretrained_model_path'] = model_id Z train_datasetpromptr)leftrighttopbottom)start_sample_framen_sample_frameZ sampling_rateoffsetz5You need to upload a video or choose a provided videonameZvalidation_sample_logger_configpromptsZ p2p_configrcross_replace_stepsZdefault_self_replace_steps Z eq_paramswordsvaluesguidance_scaleZnum_inference_stepsconfigresultz.ymlz.yaml_logdirz Saving at )rrrr r&z_0.gifz _0_0_0.mp4r)Z OmegaConfloadZget_time_stringcopydeepcopyprintupdategrError isinstancestrhasattrrsplitlenreplacetestrrrr )rmodel_id data_path source_prompt target_promptr r! enhance_wordsenhance_words_value num_stepsr%user_input_videorrstride left_crop right_croptop_crop bottom_cropZdefault_edit_configZOmegadict_default_edit_configZdataset_time_stringZ config_nowZ offset_dictZImageSequenceDataset_dictZp2p_config_nowr*Z save_pathZmp4_pathrrrrun,s^          & zmerge_config_then_run.run)rN)Nrrrrrrr)__name__ __module__ __qualname__rrFrrrrrs 2r)r8r,gradior0rrrrrs