File size: 1,109 Bytes
b14067d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
#!/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
|