diff --git a/scripts/run.sh b/scripts/run.sh index f3a9dca..374c132 100644 --- a/scripts/run.sh +++ b/scripts/run.sh @@ -8,13 +8,13 @@ config=configs/inference_$1_v1.0.yaml prompt_dir=prompts/$1/ res_dir="results" -if [ "$1" == "256" ]; then +if [ "$1" = "256" ]; then H=256 FS=3 ## This model adopts frame stride=3, range recommended: 1-6 (larger value -> larger motion) -elif [ "$1" == "512" ]; then +elif [ "$1" = "512" ]; then H=320 FS=24 ## This model adopts FPS=24, range recommended: 15-30 (smaller value -> larger motion) -elif [ "$1" == "1024" ]; then +elif [ "$1" = "1024" ]; then H=576 FS=10 ## This model adopts FPS=10, range recommended: 15-5 (smaller value -> larger motion) else @@ -22,7 +22,7 @@ else exit 1 fi -if [ "$1" == "256" ]; then +if [ "$1" = "256" ]; then CUDA_VISIBLE_DEVICES=0 python3 scripts/evaluation/inference.py \ --seed ${seed} \ --ckpt_path $ckpt \ @@ -58,4 +58,4 @@ fi ## multi-cond CFG: the is s_txt, is s_img #--multiple_cond_cfg --cfg_img 7.5 -#--loop \ No newline at end of file +#--loop diff --git a/scripts/run_application.sh b/scripts/run_application.sh index ae8943f..b48e651 100644 --- a/scripts/run_application.sh +++ b/scripts/run_application.sh @@ -8,7 +8,7 @@ res_dir="results" FS=5 ## This model adopts FPS=5, range recommended: 5-30 (smaller value -> larger motion) -if [ "$1" == "interp" ]; then +if [ "$1" = "interp" ]; then seed=12306 name=dynamicrafter_512_$1_seed${seed} CUDA_VISIBLE_DEVICES=0 python3 scripts/evaluation/inference.py \ diff --git a/scripts/run_mp.sh b/scripts/run_mp.sh index 2ae2001..86f71f7 100644 --- a/scripts/run_mp.sh +++ b/scripts/run_mp.sh @@ -9,13 +9,13 @@ config=configs/inference_$1_v1.0.yaml prompt_dir=prompts/$1/ res_dir="results" -if [ "$1" == "256" ]; then +if [ "$1" = "256" ]; then H=256 FS=3 ## This model adopts frame stride=3 -elif [ "$1" == "512" ]; then +elif [ "$1" = "512" ]; then H=320 FS=24 ## This model adopts FPS=24 -elif [ "$1" == "1024" ]; then +elif [ "$1" = "1024" ]; then H=576 FS=10 ## This model adopts FPS=10 else @@ -62,7 +62,7 @@ fi #--loop ## inference using single node with multi-GPUs: -if [ "$1" == "256" ]; then +if [ "$1" = "256" ]; then CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python3 -m torch.distributed.launch \ --nproc_per_node=8 --nnodes=1 --master_addr=127.0.0.1 --master_port=23456 --node_rank=0 \ scripts/evaluation/ddp_wrapper.py \ @@ -99,4 +99,4 @@ scripts/evaluation/ddp_wrapper.py \ --video_length 16 \ --frame_stride ${FS} \ --timestep_spacing 'uniform_trailing' --guidance_rescale 0.7 --perframe_ae -fi \ No newline at end of file +fi