|
|
#!/bin/bash |
|
|
trap 'kill 0' SIGINT |
|
|
|
|
|
export MODEL_PATH="./pretrained/CogVideoX-5b-I2V" |
|
|
|
|
|
processed_data_name=$1 |
|
|
ckpt_steps=500 |
|
|
ckpt_dir=./out/EPiC_pretrained |
|
|
ckpt_file=checkpoint-${ckpt_steps}.pt |
|
|
ckpt_path=${ckpt_dir}/${ckpt_file} |
|
|
video_root_dir="./data/${processed_data_name}" |
|
|
out_dir=${ckpt_dir}/test/${ckpt_steps}_${processed_data_name} |
|
|
|
|
|
CUDA_VISIBLE_DEVICES=0 python inference/cli_demo_camera_i2v_pcd.py \ |
|
|
--video_root_dir $video_root_dir \ |
|
|
--base_model_path $MODEL_PATH \ |
|
|
--controlnet_model_path $ckpt_path \ |
|
|
--output_path "${out_dir}" \ |
|
|
--start_camera_idx 0 \ |
|
|
--end_camera_idx 8 \ |
|
|
--controlnet_weights 1.0 \ |
|
|
--controlnet_guidance_start 0.0 \ |
|
|
--controlnet_guidance_end 0.4 \ |
|
|
--controlnet_input_channels 3 \ |
|
|
--controlnet_transformer_num_attn_heads 4 \ |
|
|
--controlnet_transformer_attention_head_dim 64 \ |
|
|
--controlnet_transformer_out_proj_dim_factor 64 \ |
|
|
--controlnet_transformer_out_proj_dim_zero_init \ |
|
|
--vae_channels 16 \ |
|
|
--num_frames 49 \ |
|
|
--controlnet_transformer_num_layers 8 \ |
|
|
--infer_with_mask \ |
|
|
--pool_style 'max' \ |
|
|
--seed 1 |
|
|
|
