diff --git a/eval/vbench_i2v/launch.sh b/eval/vbench_i2v/launch.sh index 79af0cf..d8eea1d 100644 --- a/eval/vbench_i2v/launch.sh +++ b/eval/vbench_i2v/launch.sh @@ -3,6 +3,8 @@ CKPT=$1 NUM_FRAMES=$2 MODEL_NAME=$3 +RES=$4 +ASP_RATIO=$5 if [[ $CKPT == *"ema"* ]]; then parentdir=$(dirname $CKPT) @@ -19,5 +21,10 @@ START_INDEX_LIST=(0 140 280 420 560 700 840 980) END_INDEX_LIST=(140 280 420 560 700 840 980 2000) for i in "${!GPUS[@]}"; do - CUDA_VISIBLE_DEVICES=${GPUS[i]} bash eval/sample.sh $CKPT $NUM_FRAMES $MODEL_NAME -5 ${START_INDEX_LIST[i]} ${END_INDEX_LIST[i]}>${LOG_BASE}/${TASK_ID_LIST[i]}.log 2>&1 & + if [ -z ${RES} ] || [ -z ${ASP_RATIO} ] ; + then + CUDA_VISIBLE_DEVICES=${GPUS[i]} bash eval/sample.sh $CKPT $NUM_FRAMES $MODEL_NAME -5 ${START_INDEX_LIST[i]} ${END_INDEX_LIST[i]}>${LOG_BASE}/${TASK_ID_LIST[i]}.log 2>&1 & + else + CUDA_VISIBLE_DEVICES=${GPUS[i]} bash eval/sample.sh $CKPT $NUM_FRAMES $MODEL_NAME -5 ${START_INDEX_LIST[i]} ${END_INDEX_LIST[i]} ${RES} ${ASP_RATIO}>${LOG_BASE}/${TASK_ID_LIST[i]}.log 2>&1 & + fi done